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
Copy file name to clipboardExpand all lines: imagenet/README.md
+1-1
Original file line number
Diff line number
Diff line change
@@ -30,7 +30,7 @@ You should always use the NCCL backend for multi-processing distributed training
30
30
### Single node, multiple GPUs:
31
31
32
32
```bash
33
-
python main.py -a resnet50 --lr 0.01 --dist-url 'tcp://127.0.0.1:FREEPORT' --dist-backend 'nccl' --multiprocessing-distributed --world-size 1 [imagenet-folder with train and val folders]
33
+
python main.py -a resnet50 --lr 0.01 --dist-url 'tcp://127.0.0.1:FREEPORT' --dist-backend 'nccl' --multiprocessing-distributed --world-size 1 --rank 0 [imagenet-folder with train and val folders]
0 commit comments