tf.keras.mixed_precision.experimental.set_policy

Sets the global Policy.

View aliases

Compat aliases for migration

See Migration guide for more details.

`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..