2021-12-04 15:18:50 +08:00
|
|
|
|
mindspore.nn.FixedLossScaleUpdateCell
|
|
|
|
|
|
=======================================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.nn.FixedLossScaleUpdateCell(loss_scale_value)
|
|
|
|
|
|
|
2022-01-11 09:32:02 +08:00
|
|
|
|
固定损失缩放系数的神经元。
|
2021-12-04 15:18:50 +08:00
|
|
|
|
|
2022-02-11 15:23:20 +08:00
|
|
|
|
该类是 :class:`mindspore.FixedLossScaleManager` 的 `get_update_cell` 方法的返回值。训练过程中，类 :class:`mindspore.nn.TrainOneStepWithLossScaleCell` 会调用该Cell。
|
2021-12-04 15:18:50 +08:00
|
|
|
|
|
2021-12-05 16:06:57 +08:00
|
|
|
|
**参数：**
|
2021-12-04 15:18:50 +08:00
|
|
|
|
|
2022-01-11 09:32:02 +08:00
|
|
|
|
- **loss_scale_value** (float) - 初始损失缩放系数。
|
2021-12-04 15:18:50 +08:00
|
|
|
|
|
|
|
|
|
|
**输入：**
|
|
|
|
|
|
|
2022-01-11 09:32:02 +08:00
|
|
|
|
- **loss_scale** (Tensor) - 训练期间的损失缩放系数，是一个标量。在当前类中，该值被忽略。
|
2021-12-04 20:36:47 +08:00
|
|
|
|
- **overflow** (bool) - 是否发生溢出。
|
2021-12-04 15:18:50 +08:00
|
|
|
|
|
|
|
|
|
|
**输出：**
|
|
|
|
|
|
|
|
|
|
|
|
Bool，即输入 `overflow`。
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
.. py:method:: get_loss_scale()
|
|
|
|
|
|
|
2022-01-11 09:32:02 +08:00
|
|
|
|
获取当前损失缩放系数。
|
