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

tf.keras.mixed_precision.experimental.set_policy

TensorFlow 1 version 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..