tf.train.experimental.disable_mixed_precision_graph_rewrite

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

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