TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update docs

Browse Source
This commit is contained in:
CarloLucibello 2018-07-16 07:32:13 +02:00
parent 185e9148b6
commit 071dcdda87
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/layers/normalise.jl
View File

@ -1,6 +1,5 @@
""" """
testmode!(m) testmode!(m, val=true)
testmode!(m, false)
Put layers like [`Dropout`](@ref) and [`BatchNorm`](@ref) into testing mode Put layers like [`Dropout`](@ref) and [`BatchNorm`](@ref) into testing mode
(or back to training mode with `false`). (or back to training mode with `false`).
@ -94,7 +93,11 @@ m = Chain(
Dense(64, 10), Dense(64, 10),
BatchNorm(10), BatchNorm(10),
softmax) softmax)
y = m(rand(28^2, 10))
``` ```
To use the layer at test time set [`testmode!(m, true)`](@ref).
""" """
mutable struct BatchNorm mutable struct BatchNorm
λ # activation function λ # activation function