View source on GitHub |
Library for mixed precision training.
Classes
class ExponentialUpdateLossScaleManager
: Loss scale manager uses an exponential update strategy.
class FixedLossScaleManager
: Loss scale manager with a fixed loss scale.
class LossScaleManager
: Abstract loss scale manager class.
class LossScaleOptimizer
: An optimizer that applies loss scaling in backprop.