Skip to content

Commit 635e64c

Browse files
nsvanirudhsoumith
authored andcommitted
Add missing rank argument (#490)
1 parent e0929a4 commit 635e64c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

imagenet/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ You should always use the NCCL backend for multi-processing distributed training
3030
### Single node, multiple GPUs:
3131

3232
```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]
3434
```
3535

3636
### Multiple nodes:

0 commit comments

Comments
 (0)