File size: 400 Bytes
40d87dd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | #!/usr/bin/env python3
"""
Python wrapper to launch dflash training script via northjob/torchrun
"""
import subprocess
import sys
import os
if __name__ == "__main__":
bash_script = "/workspace/hanrui/syxin_old/run_train_multinode_dflash.sh"
args = sys.argv[1:]
cmd = ["bash", bash_script] + args
result = subprocess.run(cmd, env=os.environ.copy())
sys.exit(result.returncode)
|