예제 #1
0
def get_parser():
    parser = argparse.ArgumentParser()
    parser.add_argument('--fpath', type=str)
    parser.add_argument('--len', '-l', type=int, default=0)
    parser.add_argument('--norender', '-nr', action='store_true')
    parser.add_argument('--itr', '-i', type=int, default=-1)
    parser.add_argument('--deterministic', '-d', action='store_true')
    parser.add_argument('--episode_ids', '-id', type=str, default="0")
    parser.add_argument('--bests', type=str2bool, default=None)
    parser.add_argument('--fixed_test_set', '-ts', type=str, default=None)
    parser.add_argument('--load_env', action='store_true')
    parser.add_argument('--use_test_env', action='store_true')

    parser.add_argument('--record', type=str2bool, default=False)
    parser.add_argument('--recording_path', type=str, default=None)
    EnvironmentArgsHandler.set_parser_arguments(parser)
    StudentArgsHandler.set_parser_arguments(parser)

    return parser
예제 #2
0
            "walker_type": ["old_classic_bipedal"],
            "movable_creepers": None,
        },
        # {
        #     'env':'parametric-continuous-flat-parkour-v0',
        #     "walker_type": ["classic_bipedal", "fish"],
        #     "motors_torque": 80,
        #     "water_level": 4,
        #     "dummy_param": [0, 1]
        # }
    ]

    parser = argparse.ArgumentParser()
    parser.add_argument('--seed', '-s', type=int, default=0)
    StudentArgsHandler.set_parser_arguments(parser)
    EnvironmentArgsHandler.set_parser_arguments(parser)
    TeacherArgsHandler.set_parser_arguments(parser)
    for _env in envs:
        _env["teacher"] = 'Random'
        _env["nb_test_episodes"] = 1
        _env["seed"] = 43
        current_args_dict = deepcopy(_env)
        for walker_type in _env["walker_type"]:
            print("##### Benchmarking {0} with {1} body #####".format(
                _env['env'], walker_type))
            current_args_dict["walker_type"] = walker_type
            args_str = dict_to_args_str(current_args_dict)
            args = parser.parse_args(args_str)

            for j in range(2):
                if j == 0: