Python 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 位計算 dtype 和 float32 變量 dtype。

隻有浮點策略可以設置為全局策略，例如 'float32' 和 'mixed_float16' 。 'int32' 和'complex64' 等非浮點策略不能設置為全局策略，因為大多數層不支持此類策略。

有關詳細信息，請參閱tf.keras.mixed_precision.Policy。