ViT-pytorch icon indicating copy to clipboard operation
ViT-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon jeonsworld

For torch.distributed.launch ARG --local_rank should be --local-rank

Open hassanbabaie opened this issue 10 months ago • 0 comments

When using -m torch.distributed.launch --nproc_per_node=2 as an example it uses --local-rank and not the expected --local_rank which generate a fatal error.

https://github.com/jeonsworld/ViT-pytorch/blob/460a162767de1722a014ed2261463dbbc01196b6/train.py#L281

Setting the arg to local-rank in train.py resolved the issue and it then works as expected

hassanbabaie avatar Oct 24 '23 23:10 hassanbabaie