tf.train.experimental.disable_mixed_precision_graph_rewrite

Отключает переписывание графа с использованием смешанной точности.

tf.train.experimental.disable_mixed_precision_graph_rewrite()

После вызова этой функции переписывание графа со смешанной точностью больше не будет выполняться для tf.functions, и операции с плавающей запятой одинарной точности (float32) больше не будут конвертироваться в плавающую запятую двойной точности (float16).

Это не отменяет эффекты масштабирования потерь. Любые оптимизаторы, обернутые LossScaleOptimizer, будут продолжать выполнять масштабирование потерь, хотя это масштабирование больше не будет полезно, так как переписывание графа больше не преобразует tf.functions для использования float16.

Эта функция полезна для тестирования на единицах. Тест на единицах может использовать переписывание графа со смешанной точностью, затем отключить его, чтобы будущие тесты на единицах продолжали использовать float32.