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, продолжат масштабировать потери, хотя это масштабирование больше не будет полезным, если оптимизатор используется в новых сеансах, так как переписывание графа больше не преобразует граф для использования float16.

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