You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hello, I used the following command to train
"python -m torch.distributed.launch --nproc_per_node 8 --master_port 12345 main.py --cfg configs/swin/swin_tiny_patch4_window7_224.yaml --batch-size 128 --fused_window_process --fused_layernorm --optim=fused_adam --data-path /home/datasets/imagenet"
but got "Swin Transformer training and evaluation script: error: the following arguments are required: --local_rank",how can I fix this problem, thanks
The text was updated successfully, but these errors were encountered:
I have resolved this issue and have conducted distributed training tests on both PyTorch 1.8 and 1.10 versions. I mainly made some code deletions to main.py and config.py. Specific:
'''python
Delete lines 72-73 of code from main.py
Delete lines 341-344 of code from config.py, then add config.LOCAL-RANK=int (os. environment ['LOCAL-RANK '])
'''
The above is my solution, I hope it will be useful to you.
Hello, I used the following command to train
"python -m torch.distributed.launch --nproc_per_node 8 --master_port 12345 main.py --cfg configs/swin/swin_tiny_patch4_window7_224.yaml --batch-size 128 --fused_window_process --fused_layernorm --optim=fused_adam --data-path /home/datasets/imagenet"
but got "Swin Transformer training and evaluation script: error: the following arguments are required: --local_rank",how can I fix this problem, thanks
The text was updated successfully, but these errors were encountered: