diff --git a/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train.py b/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train.py
index d65b4a85de1cb73817d00554e4f9879a213038ba..892035a051f01eb66bfb8715922c41ad20d1175c 100644
--- a/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train.py
+++ b/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train.py
@@ -42,7 +42,8 @@ from fn_utils import infolog
 import synthesize
 from utils import *
 from tensorflow.core.protobuf.rewriter_config_pb2 import RewriterConfig
-
+tf.enable_control_flow_v2()
+tf.enable_resource_variables()
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 class Graph():