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