facebookresearch / FAMBench

Benchmarks to capture important workloads.
Apache License 2.0
28 stars 23 forks source link

torch.distribute.launch -> torchrun #115

Open zstreet87 opened 1 year ago

zstreet87 commented 1 year ago

Need to replace the deprecated distribute wrapper found here https://github.com/facebookresearch/FAMBench/blob/main/benchmarks/bevt/ootb/run_bevt_train.sh#L52 with torchrun.

See here: https://pytorch.org/docs/stable/distributed.html

The deprecated command leads to a syntax error in BEVT, sending in --local-rank but expects --local_rank