diff --git a/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_full_1p.sh b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_full_1p.sh index 52985ac179f70cee58d1f4e269bf6256a3a1421c..072843e4cf449089539a66e6b8dd6c8be7f77695 100644 --- a/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_full_1p.sh +++ b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_full_1p.sh @@ -28,7 +28,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_full_8p.sh b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_full_8p.sh index add8c9ef047cfd9bd4d9d475de7364dac0102248..fb2b682946b80eebcddcd25e39e37e283fbac39a 100644 --- a/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_full_8p.sh +++ b/TensorFlow/built-in/cv/image_classification/ShufflenetV2_ID0371_For_Tensorflow/test/train_full_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