tf.keras.dtype_policies.get

Возвращает экземпляр Keras DTypePolicy.

tf.keras.dtype_policies.get(
    identifier
)

identifier может быть строковым именем класса DTypePolicy.

policy = dtype_policies.get("mixed_bfloat16")
type(loss)
<class '...FloatDTypePolicy'>

Также можно передать этой функции config политики типа данных, передав словарь, содержащий class_name и config в качестве идентификатора. Обратите внимание, что class_name должен соответствовать классу DTypePolicy.

identifier = {"class_name": "FloatDTypePolicy",
              "config": {"name": "float32"} }
policy = dtype_policies.get(identifier)
type(loss)
<class '...FloatDTypePolicy'>