Watch talks from the 2019 TensorFlow Dev Summit Watch now

Module: tf.contrib.mixed_precision

Defined in tensorflow/contrib/mixed_precision/__init__.py.

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.