burn
0.14.0
Module momentum
Structs
In burn::optim
Module
burn
::
optim
::
momentum
Copy item path
[
−
]
Expand description
Momentum module for optimizers.
Structs
ยง
Momentum
Momemtum implementation that transforms gradients.
MomentumConfig
Configuration to create
momentum
.
MomentumState
State of
momentum
.
MomentumStateItem
The record item type for the module.