trainmode!
function
defined in module
Flux
trainmode!(model) -> model
Set a layer, or all layers in a model, to training mode. Opposite to
testmode!
, see further details there.
trainmode!(m, active)
This two-argument method is deprecated.
Possible values of
active
are:
true
for training, or
false
for testing, same as
testmode!
(m)
:auto
or
nothing
for Flux to detect training automatically.
There are
4
methods for Flux.trainmode!
:
The following pages link back here: