From 314b62e26196c15388be8239e60b0dc7c4ac9de6 Mon Sep 17 00:00:00 2001 From: Carlo Lucibello Date: Wed, 4 Dec 2024 14:28:23 +0100 Subject: [PATCH] docs --- docs/src/reference/models/layers.md | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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! ```