Run `torch.distributed.launch` with `-m` option for the script

[LeviViana]

Take a look at this snippet, it could help.