tf.keras.config.set_dtype_policy

Sets the default dtype policy globally.

Main aliases

tf.keras.config.set_dtype_policy(
    policy
)

Used in the guide	Used in the tutorials
Mixed precision	Image classification with Model Garden Instance Segmentation with Model Garden Object detection with Model Garden Semantic Segmentation with Model Garden

keras.config.set_dtype_policy("mixed_float16")

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/config/set_dtype_policy