From adae6417b6f6b9e2c31da5ff0acf89f4ced7b8f8 Mon Sep 17 00:00:00 2001
From: amshaker <abdelrahman.m.shaker@gmail.com>
Date: Wed, 26 Jul 2023 01:11:47 +0400
Subject: [PATCH] update dist_train.sh

---
 dist_train.sh | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/dist_train.sh b/dist_train.sh
index f6cebbb..2578b11 100644
--- a/dist_train.sh
+++ b/dist_train.sh
@@ -4,18 +4,18 @@
 IMAGENET_PATH=$1
 nGPUs=$2
 
-## SwiftFormer-XS
+## SwiftFormer-XS training
 python -m torch.distributed.launch --nproc_per_node=$nGPUs --use_env main.py --model SwiftFormer_XS --aa="" --mixup 0 --cutmix 0 --data-path "$IMAGENET_PATH" \
 --output_dir SwiftFormer_XS_results
 
-## SwiftFormer-S
+## SwiftFormer-S training
 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
+## SwiftFormer-L1 training
 python -m torch.distributed.launch --nproc_per_node=$nGPUs --use_env main.py --model SwiftFormer_L1 --data-path "$IMAGENET_PATH" \
 --output_dir SwiftFormer_L1_results
 
-## SwiftFormer-L3
+## SwiftFormer-L3 training
 python -m torch.distributed.launch --nproc_per_node=$nGPUs --use_env main.py --model SwiftFormer_L3 --data-path "$IMAGENET_PATH" \
 --output_dir SwiftFormer_L3_results