diff --git a/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/train.py b/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/train.py
index 9402154de57d35ca21d1942fa75d56e99560ceb6..9fdc9b34e8a921e006f19e303db69325e84fe114 100644
--- a/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/train.py
+++ b/TensorFlow/built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/train.py
@@ -260,6 +260,7 @@ if args.save_optimizer:
 config = tf.ConfigProto()
 custom_op = config.graph_options.rewrite_options.custom_optimizers.add()
 custom_op.name = "NpuOptimizer"
+custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("allow_mix_precision")
 custom_op.parameter_map["use_off_line"].b = True  # training on Ascend chips
 custom_op.parameter_map["enable_data_pre_proc"].b = True
 custom_op.parameter_map["iterations_per_loop"].i = args.iterations_per_loop