tf.keras.mixed_precision.experimental.set_policy

Sets the global Policy.

View aliases

Compat aliases for migration

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

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/mixed_precision/experimental/set_policy