20 lines
966 B
ReStructuredText
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) - 指定了当前网络参数组的特定的优化器配置。 |