trainmode!
function
defined in module
Flux
trainmode!(m, mode = true)
Set a layer of model's train mode (see below). Symmetric to
testmode!
(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.
Possible values include:
true
for training
false
for testing
:auto
or
nothing
for Flux to detect the mode automatically
There is
1
method for Flux.trainmode!
:
The following pages link back here: