Loss scale manager abstract class.
Derive FixedLossScaleManager and DynamicLossScaleManager that override all LossScaleManager’s method.
-
get_loss_scale()[source]
Get loss scale value.
-
get_update_cell()[source]
Get the loss scaling update logic cell.
-
update_loss_scale(overflow)[source]
Update loss scale value.
- Parameters
overflow (bool) – Whether it overflows.