From ef5daec20c3b82a28b85dfbc1cc84d15fff3065d Mon Sep 17 00:00:00 2001
From: Abdelrahman Shaker <108531886+Amshaker@users.noreply.github.com>
Date: Wed, 26 Jul 2023 01:20:31 +0400
Subject: [PATCH] Update slurm_train.sh

---
 slurm_train.sh | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/slurm_train.sh b/slurm_train.sh
index 17c4650..85f485a 100644
--- a/slurm_train.sh
+++ b/slurm_train.sh
@@ -15,9 +15,8 @@ srun python main.py --model "$MODEL" \
 --data-path "$IMAGENET_PATH" \
 --batch-size 128 \
 --epochs 300 \
---aa="" --mixup 0 --cutmix 0
 
 
-## Note: Disable aa, mixup, and cutmix for SwiftFormer-XS only
+## Note: Disable aa, mixup, and cutmix for SwiftFormer-XS, and disable mixup, and cutmix for SwiftFormer-S.
 ## By default, this script requests total 16 GPUs on 4 nodes. The batch size per gpu is set to 128,
-## tha sums to 128*16=2048 in total.
\ No newline at end of file
+## tha sums to 128*16=2048 in total.