tf.keras.backend.set_floatx
Задает тип данных с плавающей запятой по умолчанию.
tf.keras.backend.set_floatx(
value
)
Примечание: Не рекомендуется задавать значение"float16"при обучении, так как это, вероятно, приведет к проблемам с численной стабильностью. Вместо этого используйте смешанную точность, которая сочетаетfloat16иfloat32. Ее можно настроить, вызвавkeras.mixed_precision.set_dtype_policy('mixed_float16').
| Аргументы | |
|---|---|
value | Строка; 'bfloat16', 'float16', 'float32' или 'float64'. |
Примеры:
keras.config.floatx() 'float32'
keras.config.set_floatx('float64')
keras.config.floatx()
'float64'# Set it back to float32
keras.config.set_floatx('float32')| Вызывает исключение | |
|---|---|
ValueError | Если указано недопустимое значение. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/backend/set_floatx