tf.keras.config.set_dtype_policy

Sets the default dtype policy globally.

View aliases

Main aliases

tf.keras.mixed_precision.set_dtype_policy, tf.keras.mixed_precision.set_global_policy

tf.keras.config.set_dtype_policy(
    policy
)

Used in the notebooks

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

Example:

keras.config.set_dtype_policy("mixed_float16")