24 lines
624 B
Python
24 lines
624 B
Python
import sys
|
|
import subprocess
|
|
from pybullet_utils.arg_parser import ArgParser
|
|
from pybullet_utils.logger import Logger
|
|
|
|
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)
|
|
|
|
Logger.print2('Running with {:d} workers'.format(num_workers))
|
|
cmd = 'mpiexec -n {:d} python3 DeepMimic_Optimizer.py '.format(num_workers)
|
|
cmd += ' '.join(args)
|
|
Logger.print2('cmd: ' + cmd)
|
|
subprocess.call(cmd, shell=True)
|
|
return
|
|
|
|
if __name__ == '__main__':
|
|
main()
|