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

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