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