Change torch.distributed.launch to torchrun for RNNT

facebookresearch / FAMBench

Benchmarks to capture important workloads.

Apache License 2.0

28 stars 23 forks source link

Change torch.distributed.launch to torchrun for RNNT #118

Closed pnunna93 closed 1 year ago

pnunna93 commented 1 year ago

This PR has changes to move from torch.distributed.launch to torchrun for RNNT. torch.distributed.launch is deprecated and could cause errors in the future, this change helps fix that.