tf.keras.mixed_precision.experimental.set_policy

TensorFlow 2 version View source on GitHub

Sets the global Policy.

Aliases:

tf.keras.mixed_precision.experimental.set_policy(policy)

The global policy is the default policy used for layers, if no policy is passed to the layer constructor. If no global policy is set, layers will instead default to a Policy constructed from tf.keras.backend.floatx() in TensorFlow 2. In TensorFlow 1, layers default to an "infer" policy.

See keras.mixed_precision.experimental.Policy for more information.

Args:

  • policy: A Policy, or a string that will be converted to a Policy..