TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add "during X phase" phrasing to testmode!/trainmode! docstring.

This commit is contained in:
Kyle Daruwalla 2020-03-01 12:49:30 -06:00
parent 35e460b044
commit 23f791e32b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/functor.jl
View File

@ -46,7 +46,7 @@ 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.
it back into train mode during training phase.
Possible values include:
- `false` for training
@ -62,7 +62,7 @@ Set a layer of model's train mode (see below).
Symmetric to [`testmode!`](@ref) (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.
it into test mode during testing phase.
Possible values include:
- `true` for training