tf.compat.v1.mixed_precision.disable_mixed_precision_graph_rewrite

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

tf.compat.v1.mixed_precision.disable_mixed_precision_graph_rewrite()

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

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

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