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 df5bad95c11dd24fc24649d86dd680cc9f6ced4c..3f97d3f13c07f33ef1ff8811e48e0d7dbc344b85 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,