tf.keras.mixed_precision.set_global_policy

Устанавливает глобальную политику dtype.

tf.keras.mixed_precision.set_global_policy(
    policy
)

Глобальная политика — это по умолчанию tf.keras.mixed_precision.Policy, используемая для слоёв, если политика не передана в конструктор слоя.

tf.keras.mixed_precision.set_global_policy('mixed_float16')
tf.keras.mixed_precision.global_policy()
<Policy "mixed_float16">
tf.keras.layers.Dense(10).dtype_policy
<Policy "mixed_float16">
# Global policy is not used if a policy is directly passed to constructor
tf.keras.layers.Dense(10, dtype='float64').dtype_policy
<Policy "float64">
tf.keras.mixed_precision.set_global_policy('float32')

Если глобальная политика не установлена, слои будут по умолчанию использовать политику, построенную из tf.keras.backend.floatx().

Для использования смешанной точности глобальная политика должна быть установлена в 'mixed_float16' или 'mixed_bfloat16', чтобы каждый слой по умолчанию использовал 16-битный тип вычислений и тип переменных float32.

В качестве глобальной политики могут быть установлены только политики с плавающей точкой, такие как 'float32' и 'mixed_float16'. Политики без плавающей точки, такие как 'int32' и 'complex64', не могут быть установлены в качестве глобальной политики, так как большинство слоёв не поддерживают такие политики.

См. tf.keras.mixed_precision.Policy для получения дополнительной информации.