From 311344d7ddb0e12aee10e5dd1e36ad3fe2629532 Mon Sep 17 00:00:00 2001 From: huangju1993 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