File size: 217 Bytes
455d74c | 1 2 3 4 5 6 7 8 | #!/home/crhon/pytorch_rocm_env/bin/python3
import sys
from torch.distributed.run import main
if __name__ == '__main__':
if sys.argv[0].endswith('.exe'):
sys.argv[0] = sys.argv[0][:-4]
sys.exit(main())
|