tf.keras.mixed_precision.experimental.set_policy

Stay organized with collections Save and categorize content based on your preferences.

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