Skip to content

Commit 44053c5

Browse files
daquexiansoumith
authored andcommitted
Add missing world size argument (#484)
15e2771 makes the command of single node fail
1 parent d6b5211 commit 44053c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

imagenet/README.md

Lines changed: 1 addition & 1 deletion
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 [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 [imagenet-folder with train and val folders]
3434
```
3535

3636
### Multiple nodes:

0 commit comments

Comments
 (0)