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.print('Running with {:d} workers'.format(num_workers)) cmd = 'mpiexec -n {:d} python DeepMimic_Optimizer.py '.format(num_workers) cmd += ' '.join(args) Logger.print('cmd: ' + cmd) subprocess.call(cmd) return
def build_arg_parser(args): arg_parser = ArgParser() arg_parser.load_args(args) arg_file = arg_parser.parse_string('arg_file', '') if (arg_file != ''): succ = arg_parser.load_file(arg_file) assert succ, Logger.print('Failed to load args from: ' + arg_file) rand_seed_key = 'rand_seed' if (arg_parser.has_key(rand_seed_key)): rand_seed = arg_parser.parse_int(rand_seed_key) rand_seed += 1000 * MPIUtil.get_proc_rank() Util.set_global_seeds(rand_seed) return arg_parser
def main(): global args global world # Command line arguments args = sys.argv[1:] arg_parser = ArgParser() arg_parser.load_args(args) max_iters = arg_parser.parse_int('max_iters', None) world = build_world(args, enable_draw=False) run(max_iters) shutdown() return
def build_arg_parser(args): # use arguments and determine from txt arg_parser = ArgParser() # store a _table using dict arg_parser.load_args(args) # ensure arg_file is exist, and load the file in it arg_file = arg_parser.parse_string('arg_file', '') if (arg_file != ''): # read the arguments in arg_file.txt and store in _table succ = arg_parser.load_file(arg_file) assert succ, Logger.print('Failed to load args from: ' + arg_file) rand_seed_key = 'rand_seed' if (arg_parser.has_key(rand_seed_key)): rand_seed = arg_parser.parse_int(rand_seed_key) rand_seed += 1000 * MPIUtil.get_proc_rank() Util.set_global_seeds(rand_seed) return arg_parser