cluster_gym_stub.py 文件源码

def main(argv=None):
    if len(sys.argv) <= 2:
        print('Usage ./%s --exp=<exp_name> --ec2_settings=<relative_path_to_ec2_settings_file>'%sys.argv[0])
        sys.exit(0)
    import importlib
    import os.path as osp
    import shutil
    module_name = 'sandbox.rocky.tf.launchers.%s'%(
        osp.splitext(FLAGS.ec2_settings)[0].replace('/','.'))
    mod = importlib.import_module(module_name)
    dst_py = osp.join(osp.dirname(FLAGS.ec2_settings),FLAGS.exp+'.py')
    try:
        shutil.copy(FLAGS.ec2_settings, dst_py)
    except shutil.SameFileError as e:
        print(e)
    if type(mod.params) != list: mod.params = [mod.params]
    if hasattr(mod, 'base_params'): base_params = mod.base_params
    else: base_params = dict()
    N = 0
    for params in mod.params:
        ps = base_params.copy()
        ps.update(params)
        N += execute(params=ps, mode="ec2")
    print('Launched %d jobs.'%N)