23 lines
837 B
ReStructuredText
23 lines
837 B
ReStructuredText
mindspore.nn.OptTFTWrapper
|
||
==========================
|
||
|
||
.. py:class:: mindspore.nn.OptTFTWrapper(opt, **kwargs)
|
||
|
||
实现TFT优化器封装器。该封装器将在优化器更新前向MindIO TFT上报状态。
|
||
|
||
.. note::
|
||
该优化器依赖于MindIO TFT特性。当前只支持Ascend后端的图模式，并且sink_size的配置必须小于等于1。
|
||
|
||
参数：
|
||
- **opt** (Optimizer) - 该参数必须为Optimizer的子类。
|
||
|
||
输入：
|
||
- **gradients** (tuple[Tensor]) - 参数opt的 `params` 的梯度，shape与opt的 `params` shape 相同。
|
||
|
||
输出：
|
||
Tensor，优化器opt执行返回的结果。
|
||
|
||
异常：
|
||
- **TypeError** - 如果opt不是Optimizer的子类。
|
||
- **ValueError** - 如果不是运行在Ascend后端的图模式，或者用户不开启TFT特性。
|