From 5002a884b362b1dd4bb068d7f3ee5282a2307594 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=86=E6=96=87=E9=9C=9E?=
 <10511946+lu-wenxia@user.noreply.gitee.com>
Date: Fri, 15 Apr 2022 03:27:29 +0000
Subject: [PATCH] update
 TensorFlow/contrib/cv/EfficientNet/EfficientNet_V2_ID1220_for_TensorFlow/train.py.

---
 .../EfficientNet/EfficientNet_V2_ID1220_for_TensorFlow/train.py  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/TensorFlow/contrib/cv/EfficientNet/EfficientNet_V2_ID1220_for_TensorFlow/train.py b/TensorFlow/contrib/cv/EfficientNet/EfficientNet_V2_ID1220_for_TensorFlow/train.py
index e5a165d46..39dcda7b6 100644
--- a/TensorFlow/contrib/cv/EfficientNet/EfficientNet_V2_ID1220_for_TensorFlow/train.py
+++ b/TensorFlow/contrib/cv/EfficientNet/EfficientNet_V2_ID1220_for_TensorFlow/train.py
@@ -184,6 +184,7 @@ config = tf.ConfigProto(allow_soft_placement=True)
 custom_op = config.graph_options.rewrite_options.custom_optimizers.add()
 custom_op.name = "NpuOptimizer"
 custom_op.parameter_map["use_off_line"].b = True  # 在昇腾AI处理器执行训练
+custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes("allow_mix_precision")
 config.graph_options.rewrite_options.remapping = RewriterConfig.OFF  # 关闭remap开关
 sess = tf.Session(config=config)
 sess.run(tf.global_variables_initializer())
-- 
Gitee