TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tf.contrib.mixed_precision

View source on GitHub

Library for mixed precision training.


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.