tf.compat.v2.train.experimental.disable_mixed_precision_graph_rewrite

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

tf.compat.v2.train.experimental.disable_mixed_precision_graph_rewrite()

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

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

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