diff --git a/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py b/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py
index 45c0c428db8ece0ccc3fcb990d347ecd26e9e75c..c777d09f270a71d986fa13a57b7fc2f59055881f 100644
--- a/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py
+++ b/TensorFlow/built-in/cv/image_synthesis/ViDeNN_ID1146_for_TensorFlow/main_spatialCNN.py
@@ -115,6 +115,7 @@ def main(_):
 	custom_op_ = global_config_static.graph_options.rewrite_options.custom_optimizers.add()
 	custom_op_.name = 'NpuOptimizer'
 	custom_op_.parameter_map["use_off_line"].b = True
+    custom_op_.parameter_map["graph_run_mode"].i = 0
 	global_config_static.graph_options.rewrite_options.remapping = RewriterConfig.OFF
 
 	#npu modify end