Skip to content

Commit

Permalink
Revert BatchNorm example
Browse files Browse the repository at this point in the history
  • Loading branch information
Saransh-cpp committed Jun 11, 2022
1 parent 9d48e51 commit 91fd655
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions src/layers/normalise.jl
Original file line number Diff line number Diff line change
Expand Up @@ -255,13 +255,13 @@ that will be used to renormalize the input in test phase.
Use [`testmode!`](@ref) during inference.
# Examples
```jldoctest
julia> m = Chain(
Dense(28^2 => 64),
BatchNorm(64, relu),
Dense(64 => 10),
BatchNorm(10),
softmax);
```julia
m = Chain(
Dense(28^2 => 64),
BatchNorm(64, relu),
Dense(64 => 10),
BatchNorm(10),
softmax)
```
"""
mutable struct BatchNorm{F,V,N,W}
Expand Down

0 comments on commit 91fd655

Please sign in to comment.