hexone2086/asmo_vhead
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update slurm_train.sh

Browse Source
This commit is contained in:
Abdelrahman Shaker
2023-07-26 01:20:31 +04:00
committed by GitHub
parent 49bf3f55f0
commit ef5daec20c
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
slurm_train.sh
View File

@@ -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.