tf.keras.mixed_precision.experimental.set_policy

TensorFlow 2 version View source on GitHub

Sets the global 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.

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