megengine.optimizer

Optimizer

所有优化器的基类。

常见优化器

SGD

实现随机梯度下降。

AdamW

Implements AdamW algorithm proposed in “Decoupled Weight Decay Regularization”.

Adam

实现 “Adam: A Method for Stochastic Optimization” 中提出的Adam算法。

Adagrad

实现Adagrad算法。

Adadelta

实现Adadelta算法。

学习率调整

LRScheduler

所有学习率调度器的基类。

MultiStepLR

以gamma为倍率阶梯式衰减各参数组的学习率

梯度处理

clip_grad_norm

Clips gradient norm of an iterable of parameters.

clip_grad_value

Clips gradient of an iterable of parameters to a specified lower and upper.