diff --git a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_1p_profiling2.sh b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_1p_profiling2.sh index 6823d590634f09b4876755b02533dad8cc103671..31313feeb22fc7b1e798e5f00795d47bfbad2c4e 100644 --- a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_1p_profiling2.sh +++ b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_performance_1p_profiling2.sh @@ -17,11 +17,11 @@ data_path="" #网络名称,同目录名称 Network="ResNet50_ID0360_for_TensorFlow2.X" #训练epoch -train_epochs=2 +train_epochs=1 #训练batch_size batch_size=256 #训练step -train_steps=`expr 1281167 / ${batch_size}` +train_steps=`expr 128116 / ${batch_size}` #学习率 learning_rate=0.495