diff --git a/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh b/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh
index af76da2a5e6c0f6a844d1d34773df719b5e56157..3331e037e5ee93655a0746ff4fb3c9f40252aeef 100755
--- a/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh
+++ b/cv/classification/resnest50/pytorch/train_resnest50_amp_dist.sh
@@ -26,5 +26,5 @@ fi
 cd ${ROOT_DIR}
 python3  $PYTHONARG ${ROOT_DIR}/run_train.py \
 	--model resnest50 --dali --dali-cpu  --data-path $DATA_PATH \
-	--opt fused_sgd --batch-size 512 --lr 0.0125  \
+	--opt fused_sgd --batch-size 256 --lr 0.0125  \
 	--amp --nhwc "$@"