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

update dis_train.sh

Browse Source
This commit is contained in:
amshaker
2023-07-26 01:05:02 +04:00
parent 670dea3e1f
commit ff08bf624d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dist_train.sh
View File

@@ -9,7 +9,7 @@ python -m torch.distributed.launch --nproc_per_node=$nGPUs --use_env main.py --m
--output_dir SwiftFormer_XS_results --output_dir SwiftFormer_XS_results
## SwiftFormer-S ## 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 --output_dir SwiftFormer_S_results
## SwiftFormer-L1 ## SwiftFormer-L1