tf.keras.mixed_precision.set_global_policy

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

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.