diff --git a/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/test/train_RT2_performance_1p.sh b/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/test/train_RT2_performance_1p.sh
index 23d5e23c73be818d3351b4eb138aced421a0fd71..06a913a30f28650dc59c5001360704b0c339282f 100644
--- a/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/test/train_RT2_performance_1p.sh
+++ b/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/test/train_RT2_performance_1p.sh
@@ -12,6 +12,8 @@ RANK_ID_START=0
 #使能RT2.0
 export ENABLE_RUNTIME_V2=1
 
+export ENABLE_FORCE_V2_CONTROL=1
+
 # 数据集路径,保持为空,不需要修改
 data_path=""
 
diff --git a/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train_rt.py b/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train_rt.py
index 748c4ac102d2c07c7546a79317c4d5fb163ae69f..76db3f7b5c00f38f428dcff2f0d85a2f19a4c2ba 100644
--- a/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train_rt.py
+++ b/TensorFlow/built-in/audio/Tacotron2-v1-Encoder_ID1997_for_TensorFlow/train_rt.py
@@ -232,6 +232,7 @@ def main():
 
     session_config.graph_options.rewrite_options.remapping = RewriterConfig.OFF
     session_config.graph_options.rewrite_options.memory_optimization = RewriterConfig.OFF
+    session_config.graph_options.rewrite_options.function_optimization = RewriterConfig.OFF
     npu_config = npu_config_proto(config_proto=session_config)
 
     train_dataset, num_batch = make_dataset(config, hp.train_form)