text: Title: PyTorch With distributed.sh train.py: error: unrecognized arguments: --local-rank=0
  Body: **Describe the bug**
  PyTorch With distributed.sh train.py: error: unrecognized arguments: --local-rank=0
  FutureWarning: The module torch.distributed.launch is deprecated
  and will be removed in future. Use torchrun.
  Note that --use-env is set by default in torchrun
  **To Reproduce**
  Steps to reproduce the behavior:
  1. ./distributed_train.sh
  2.
  
  **Expected behavior**
  A clear and concise description of what you expected to happen.
  
  **Screenshots**
  If applicable, add screenshots to help explain your problem.
  ![image](https://user-images.githubusercontent.com/78899198/226335714-c2b11bb3-7110-4cf8-9fd9-d05501e0de8e.png)
  
  
  **Additional context**
  I solved it by use torchrun --nproc_per_node=$NUM_PROC train.py 
  