testmode!
function
defined in module
Flux
testmode!(m, mode = true)
Set a layer or model's test mode (see below). Using
:auto
mode will treat any gradient computation as training.
Note: if you manually set a model into test mode, you need to manually place it back into train mode during training phase.
Possible values include:
false
for training
true
for testing
:auto
or
nothing
for Flux to detect the mode automatically
There are
6
methods for Flux.testmode!
:
The following pages link back here: