megengine.optimizer.Optimizer

class Optimizer(params, defaults)[源代码]

所有优化器的基类。

参数

  • params (Union[Iterable[Parameter], dict]) – 指定应该优化哪些张量。

  • defaults (dict) – 一个含有优化器默认参数的dict，如含有学习率(learning rate)和动量(momentum)。

方法

add_param_group(param_group)

向 Optimizer 的 param_groups 中添加一组参数。

backward(loss)

bcast_param()

clear_grad()

把所有参数的梯度属性设置为 None。

load_state_dict(state)

加载优化器状态。

state_dict([keep_var])

导出优化器状态。

step()

执行单一优化步骤。

zero_grad()

1.0 版后已移除.

上一页

megengine.optimizer

下一页

megengine.optimizer.Optimizer.add_param_group