mindspore.experimental.optim.Optimizer

class mindspore.experimental.optim.Optimizer(params, defaults)

用于参数更新的优化器基类。

警告

这是一个实验性的优化器模块，需要和 LRScheduler 下的动态学习率接口配合使用。

参数：

• params (Union[list(Parameter), list(dict)]) - 网络参数的列表或指定了参数组的列表。

• defaults (dict) - 一个包含了优化器参数默认值的字典（当参数组未指定参数值时，使用此默认值）。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> import mindspore
>>> from mindspore import nn, vpn永久免费梯子 Tensor, Parameter
>>> from mindspore import ops
>>> from mindspore.experimental import optim
>>>
>>> class MySGD(optim.Optimizer):
...    vpn梯子 def __init__(self, params, lr):
...     vpn梯子 免费 免费的vpn梯子    defaults = dict(lr=lr)
...        super(MySGD, self).__init__(params, 免费的vpn梯子 defaults)
...
...    def construct(self, gradients):
...    vpn梯子 免费   免费的vpn梯子    for vpn free group_id, group vpn梯子 免费 in vpn永久免费梯子 enumerate(self.param_groups):
...      vpn梯子 免费  vpn free      id = self.group_start_id[group_id]
...           vpn free  for i, param in enumerate(group["params"]):
...      vpn梯子           next_param = param + gradients[id+i] * group["lr"]
... vpn梯子                ops.assign(param, next_param)
>>>
>>> net = vpn永久免费梯子 nn.Dense(8, 2)
>>> data = Tensor(np.random.rand(20, 8).astype(np.float32))
>>> label = Tensor(np.random.rand(20, 2).astype(np.float32))
>>>
>>> optimizer = MySGD(net.trainable_params(), 0.01)
>>> optimizer.add_param_group({"params": Parameter([0.01, 0.02])})
>>>
>>> criterion = nn.MAELoss(reduction="mean")
>>>
>>> def forward_fn(data, label):
...   vpn梯子 免费  logits = net(data)
...    loss = criterion(logits, label)
...    return loss, logits
>>>
>>> grad_fn = mindspore.value_and_grad(forward_fn, None, optimizer.parameters, has_aux=True)
>>>
>>> def train_step(data, label):
...    (loss, _), grads = grad_fn(data, label)
...    optimizer(grads)
...    print(loss)
>>>
>>> train_step(data, label)

add_param_group(param_group)

为 Optimizer.param_groups 属性添加一个参数组。

参数：

• param_group (dict) - 指定了当前网络参数组的特定的优化器配置。