Added docs on testmode!
This commit is contained in:
parent
924b8f49ec
commit
ba5259a269
|
@ -66,6 +66,14 @@ LayerNorm
|
||||||
GroupNorm
|
GroupNorm
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Testmode
|
||||||
|
|
||||||
|
Many 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. Still, depending on your use case, it may be helpful to manually specify when these layers should be treated as being trained or not. For this, Flux provides `testmode!`. When called on a model (e.g. a layer or chain of layers), this function will place the model into the mode specified.
|
||||||
|
|
||||||
|
```@docs
|
||||||
|
testmode!
|
||||||
|
```
|
||||||
|
|
||||||
## Cost Functions
|
## Cost Functions
|
||||||
```@docs
|
```@docs
|
||||||
mse
|
mse
|
||||||
|
|
Loading…
Reference in New Issue