diff --git a/examples/image-classification/README.md b/examples/image-classification/README.md
index 4253210f08..747f4c38ef 100644
--- a/examples/image-classification/README.md
+++ b/examples/image-classification/README.md
@@ -241,8 +241,8 @@ python ../gaudi_spawn.py \
     --do_eval \
     --learning_rate 2e-4 \
     --num_train_epochs 5 \
-    --per_device_train_batch_size 128 \
-    --per_device_eval_batch_size 64 \
+    --per_device_train_batch_size 32 \
+    --per_device_eval_batch_size 32 \
     --eval_strategy epoch \
     --save_strategy epoch \
     --load_best_model_at_end True \