diff --git a/docs/src/reference/models/layers.md b/docs/src/reference/models/layers.md index 26c77dea91..b798a35291 100644 --- a/docs/src/reference/models/layers.md +++ b/docs/src/reference/models/layers.md @@ -145,7 +145,6 @@ Several normalisation layers behave differently under training and inference (te 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. ```@docs -testmode!(::Any) -testmode!(::Any, ::Any) +testmode! trainmode! ```