tf.compat.v1.train.experimental.disable_mixed_precision_graph_rewrite

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

tf.compat.v1.train.experimental.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 не оказывают никакого влияния на существующие сессии.