diff --git a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh index fe47982c0145e12e5617f52c9a518d42eb91ab06..e6519e7ce295f5b699a25ace303ee1636bc6af12 100644 --- a/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh +++ b/TensorFlow2/built-in/cv/image_classification/ResNet50_ID0360_for_TensorFlow2.X/test/train_full_8p_256bs_SGD.sh @@ -140,7 +140,7 @@ do --single_l2_loss_op \ --warmup_epochs=5 \ --weight_decay=0.000025 \ - --lr_schedule=polynomial \ + --lr_schedule=piecewise \ --drop_eval_remainder=True \ --precision_mode=${precision_mode} \ --over_dump=${over_dump} \