Missed TensorFlow World? Check out the recap. Learn more

tf.keras.mixed_precision.experimental.set_policy

View source on GitHub

Sets the global Policy.

Aliases:

  • tf.compat.v1.keras.mixed_precision.experimental.set_policy
  • tf.compat.v2.keras.mixed_precision.experimental.set_policy
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..