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
'env': 'parametric-continuous-parkour-v0',
"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):
import argparse
import os
from TeachMyAgent.run_utils.environment_args_handler import EnvironmentArgsHandler
from TeachMyAgent.run_utils.teacher_args_handler import TeacherArgsHandler
from TeachMyAgent.run_utils.student_args_handler import StudentArgsHandler
if __name__ == '__main__':
# Argument definition
print('Preparing the parsing...')
parser = argparse.ArgumentParser()
parser.add_argument('--exp_name', type=str, default='test')
parser.add_argument('--seed', '-s', type=int, default=0)
StudentArgsHandler.set_parser_arguments(parser)
EnvironmentArgsHandler.set_parser_arguments(parser)
TeacherArgsHandler.set_parser_arguments(parser)
# Argument parsing
args = parser.parse_args()
# Bind this run to specific GPU if there is one
if args.gpu_id is not None:
os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu_id)
print('Setting up the environment...')
env_f, param_env_bounds, initial_dist, target_dist = EnvironmentArgsHandler.get_object_from_arguments(
args)
print('Setting up the teacher algorithm...')
Teacher = TeacherArgsHandler.get_object_from_arguments(