From 524d2c8f9c7ca82690ec36ef63e5d3b46a0754d3 Mon Sep 17 00:00:00 2001 From: huangju1993 Date: Wed, 11 Jan 2023 02:39:19 +0000 Subject: [PATCH] update TensorFlow/built-in/cv/detection/CRNN_for_TensorFlow/tools/train_npu.py. Signed-off-by: huangju1993 --- .../cv/detection/CRNN_for_TensorFlow/tools/train_npu.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/TensorFlow/built-in/cv/detection/CRNN_for_TensorFlow/tools/train_npu.py b/TensorFlow/built-in/cv/detection/CRNN_for_TensorFlow/tools/train_npu.py index 30535c2c7..ba868a03c 100644 --- a/TensorFlow/built-in/cv/detection/CRNN_for_TensorFlow/tools/train_npu.py +++ b/TensorFlow/built-in/cv/detection/CRNN_for_TensorFlow/tools/train_npu.py @@ -68,7 +68,8 @@ from npu_bridge.estimator.npu.npu_estimator import NPUEstimator from npu_bridge.estimator.npu.npu_optimizer import allreduce from npu_bridge.estimator.npu.npu_optimizer import NPUDistributedOptimizer from npu_bridge.hccl import hccl_ops - +tf.enable_control_flow_v2() +tf.enable_resource_variables() CFG = global_config.cfg -- Gitee