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 30535c2c7e5db1af4ecfa32b9800f91268339143..ba868a03c1b9ec093af444afbd82532e027738e6 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