From ff08bf624df2093b227695bf31a99a9858d29045 Mon Sep 17 00:00:00 2001 From: amshaker Date: Wed, 26 Jul 2023 01:05:02 +0400 Subject: [PATCH] update dis_train.sh --- dist_train.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dist_train.sh b/dist_train.sh index 0f81d00..f6cebbb 100644 --- a/dist_train.sh +++ b/dist_train.sh @@ -9,7 +9,7 @@ python -m torch.distributed.launch --nproc_per_node=$nGPUs --use_env main.py --m --output_dir SwiftFormer_XS_results ## SwiftFormer-S -python -m torch.distributed.launch --nproc_per_node=$nGPUs --use_env main.py --model SwiftFormer_S --data-path "$IMAGENET_PATH" \ +python -m torch.distributed.launch --nproc_per_node=$nGPUs --use_env main.py --model SwiftFormer_S --mixup 0 --cutmix 0 --data-path "$IMAGENET_PATH" \ --output_dir SwiftFormer_S_results ## SwiftFormer-L1