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 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):
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 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
from pygame import display, FULLSCREEN
from engine.screensaver import ScreenSaver
from util.arg_parser import ArgParser
def run(settings):
if settings.full_screen:
screen = display.set_mode([0, 0], FULLSCREEN)
else:
screen = display.set_mode([settings.default_screen_width, settings.default_screen_height])
screen_saver = ScreenSaver(screen, settings)
screen_saver.run()
if __name__ == "__main__":
run(ArgParser.require_args())
