From 311344d7ddb0e12aee10e5dd1e36ad3fe2629532 Mon Sep 17 00:00:00 2001
From: huangju1993 <huangju1993@163.com>
Date: Thu, 31 Mar 2022 03:28:28 +0000
Subject: [PATCH] 
 built-in/cv/image_classification/EfficientNet-B4_ID0162_for_TensorFlow/main_npu.py.

---
 .../EfficientNet-B4_ID0162_for_TensorFlow/main_npu.py           | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TensorFlow/built-in/cv/image_classification/EfficientNet-B4_ID0162_for_TensorFlow/main_npu.py b/TensorFlow/built-in/cv/image_classification/EfficientNet-B4_ID0162_for_TensorFlow/main_npu.py
index df5bad95c..3f97d3f13 100644
--- a/TensorFlow/built-in/cv/image_classification/EfficientNet-B4_ID0162_for_TensorFlow/main_npu.py
+++ b/TensorFlow/built-in/cv/image_classification/EfficientNet-B4_ID0162_for_TensorFlow/main_npu.py
@@ -714,7 +714,7 @@ def main(unused_argv):
   os.environ['TF_ENABLE_WINOGRAD_NONFUSED'] = '1'
 
   config = NPURunConfig(
-    #precision_mode="allow_fp32_to_fp16",
+    precision_mode="allow_mix_precision",
     #enable_data_pre_proc=True,
     save_checkpoints_steps=FLAGS.num_train_images // (FLAGS.train_batch_size * int(os.getenv('RANK_SIZE'))),
     session_config=estimator_config,
-- 
Gitee