From c4363c808c9c3e283d66806a9cc7c19a91dd7f2e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=AE=9D=E5=BC=9F?= Date: Thu, 24 Mar 2022 06:09:07 +0000 Subject: [PATCH] update --- .../built-in/cv/detection/YoloV3_ID0076_for_TensorFlow/train.py | 1 + 1 file changed, 1 insertion(+) 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 9402154de..9fdc9b34e 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 -- Gitee