diff --git a/README.md b/README.md index cb97067..e7b0343 100644 --- a/README.md +++ b/README.md @@ -57,7 +57,7 @@ Download the pretrained models from [MAE](https://github.com/facebookresearch/ma bash tools/dist_train.sh --cfg-options model.pretrained= # for multiple machines -python -m torch.distributed.launch --nnodes --node_rank --nproc_per_node --master_addr --master_port tools/train.py --cfg-options model.pretrained= +python -m torch.distributed.launch --nnodes --node_rank --nproc_per_node --master_addr --master_port tools/train.py --cfg-options model.pretrained= --launcher pytorch ``` ## Todo