diff --git a/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_1p.sh b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_1p.sh index b22a811d789f42dc6f64d595f1392f40fc3b3bbd..3cb49e4254f04af54fc35963b97f055a43b0bdb6 100644 --- a/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_1p.sh +++ b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_1p.sh @@ -29,7 +29,7 @@ train_steps=`expr 1281167 / ${batch_size}` learning_rate= #维测参数,precision_mode需要模型审视修改 -precision_mode="" +precision_mode="allow_fp32_to_fp16" #维持参数,以下不需要修改 over_dump=False data_dump_flag=False diff --git a/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_8p.sh b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_8p.sh index 3480c969304d1466debe724199737573fd44608f..01ed6040b94435bab0ad3b816bb152d9d7eb5aa6 100644 --- a/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_8p.sh +++ b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_performance_8p.sh @@ -31,7 +31,7 @@ learning_rate="" export NPU_LOOP_SIZE=${train_steps} #维测参数,precision_mode需要模型审视修改 -precision_mode="" +precision_mode="allow_fp32_to_fp16" #维持参数,以下不需要修改 over_dump=False data_dump_flag=False