TensorFlow 2 version
|
View source on GitHub
|
Sets the global 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.. |