| #!/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) | |