tf.keras.mixed_precision.experimental.set_policy

本页内容
Args

TensorFlow 1 version

View source on GitHub

Sets the global Policy.

View aliases

Compat aliases for migration

See Migration guide for more details.

`tf.compat.v1.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..

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-10-01 UTC.

tf.keras.mixed_precision.experimental.set_policy

View aliases

Args