Modules§
- gradient_
centralization - Gradient centralization for training stabilization
Structs§
- Model
Ensemble - Model ensemble averaging
- Polyak
Averager - Polyak averaging (exponential moving average with adaptive decay)
- Weight
Averager - Weight averager for model parameters
Enums§
- Averaging
Method - Weight averaging methods