You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/src/models/layers.md
+5Lines changed: 5 additions & 0 deletions
Original file line number
Diff line number
Diff line change
@@ -127,6 +127,11 @@ Flux.dropout
127
127
128
128
Several normalisation layers behave differently under training and inference (testing). By default, Flux will automatically determine when a layer evaluation is part of training or inference.
129
129
130
+
!!! warning
131
+
This automatic train/test detection works best with Zygote, the default
132
+
automatic differentiation package. It may not work with other packages
133
+
such as Tracker, Yota, or ForwardDiff.
134
+
130
135
The functions `Flux.trainmode!` and `Flux.testmode!` let you manually specify which behaviour you want. When called on a model, they will place all layers within the model into the specified mode.
0 commit comments