diff --git a/src/functor.jl b/src/functor.jl index c97fd737..ba1df995 100644 --- a/src/functor.jl +++ b/src/functor.jl @@ -24,7 +24,7 @@ testmode!(m, mode = true) = m trainmode!(m, mode = true) Set a layer of model's train mode (see below). -Symmetric to [`testmode!`](@ref) (i.e. `trainmode!(m, mode) == testmode!(m, !mode)). +Symmetric to [`testmode!`](@ref) (i.e. `trainmode!(m, mode) == testmode!(m, !mode)`). _Note_: if you manually set a model into train mode, you need to manually place it into test mode during testing phase.