diff --git a/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py b/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py
index 4c4945735d48289fcdeaced9a3ab31685c2cc401..9a057c1e743526b3690294bbdaf49522cf4cf116 100644
--- a/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py
+++ b/TensorFlow/built-in/cv/image_classification/MobileNetV1_ID0093_for_TensorFlow/train.py
@@ -126,6 +126,7 @@ def main(args):
         custom_op.name = "NpuOptimizer"
         custom_op.parameter_map["use_off_line"].b = True
         custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes(args.precision_mode)
+        custom_op.parameter_map["graph_run_mode"].i = 0
         config_proto.graph_options.rewrite_options.remapping = RewriterConfig.OFF
         sess = tf.Session(config=npu_config_proto(config_proto=config_proto))
         sess.run(tf.global_variables_initializer())