megengine.optimizer.Optimizer¶
-
class
Optimizer
(params, defaults)[源代码]¶ 基类:
object
所有优化器的基类。
- 参数
方法
__init__
(params, defaults)初始化方法。
add_param_group
(param_group)向
Optimizer
的param_groups
中添加一组参数。backward
(loss)bcast_param
()把所有参数的梯度属性设置为 None。
load_state_dict
(state)加载优化器状态。
state_dict
([keep_var])导出优化器状态。
step
()执行单一优化步骤。
zero_grad
()1.0 版后已移除.