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'>