tf.train.experimental.disable_mixed_precision_graph_rewrite
| Просмотреть исходный код на GitHub |
Отключает переписывание графа с использованием смешанной точности. (устарело)
tf.train.experimental.disable_mixed_precision_graph_rewrite()
После вызова переписывание графа с использованием смешанной точности больше не будет выполняться для tf.functions, и операции с плавающей точкой float32 больше не будут преобразовываться в float16.
Это не отменяет эффекты масштабирования потерь. Любые оптимизаторы, обернутые с помощью LossScaleOptimizer, продолжат выполнять масштабирование потерь, хотя это масштабирование потерь больше не будет полезным, так как переписывание графа больше не преобразует tf.functions для использования float16.
Эта функция полезна для тестирования на единицах. Тест на единицу может проверить использование переписывания графа с использованием смешанной точности, а затем отключить его, чтобы последующие тесты на единицы продолжали использовать float32.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/train/experimental/disable_mixed_precision_graph_rewrite