https://github.com/xbpeng/DeepMimic
Tip revision: 70e7c6b22b775bb9342d4e15e6ef0bd91a55c6c0 authored by xbpeng on 30 April 2022, 17:36:51 UTC
Merge pull request #173 from WilliamTambellini/buildLinux
Merge pull request #173 from WilliamTambellini/buildLinux
Tip revision: 70e7c6b
mpi_run.py
import sys
import subprocess
from util.arg_parser import ArgParser
from util.logger import Logger
import DeepMimic_Optimizer
def main():
# Command line arguments
args = sys.argv[1:]
arg_parser = ArgParser()
arg_parser.load_args(args)
num_workers = arg_parser.parse_int('num_workers', 1)
assert(num_workers > 0)
if (num_workers > 1):
Logger.print('Running with {:d} workers'.format(num_workers))
cmd = 'mpiexec -n {:d} python3 DeepMimic_Optimizer.py '.format(num_workers)
cmd += ' '.join(args)
Logger.print('cmd: ' + cmd)
subprocess.call(cmd, shell=True)
else:
DeepMimic_Optimizer.main()
return
if __name__ == '__main__':
main()