mindspore/docs/api/api_python/experimental/optim/mindspore.experimental.opti...

20 lines
966 B
ReStructuredText

mindspore.experimental.optim.Optimizer
=======================================
.. py:class:: mindspore.experimental.optim.Optimizer(params, defaults)
用于参数更新的优化器基类。
.. warning::
这是一个实验性的优化器模块,需要和 `LRScheduler <https://www.mindspore.cn/docs/zh-CN/master/api_python/mindspore.experimental.html#lrscheduler%E7%B1%BB>`_ 下的动态学习率接口配合使用。
参数:
- **params** (Union[list(Parameter), list(dict)]) - 网络参数的列表或指定了参数组的列表。
- **defaults** (dict) - 一个包含了优化器参数默认值的字典(当参数组未指定参数值时,使用此默认值)。
.. py:method:: add_param_group(param_group)
`Optimizer.param_groups` 属性添加一个参数组。
参数:
- **param_group** (dict) - 指定了当前网络参数组的特定的优化器配置。