Commit 6bb81e35 authored by Benjamin Beyret's avatar Benjamin Beyret
Browse files

decreasing curriculum rewards thresholds

parent 66bc7d71
{ {
"measure": "progress", "measure": "reward",
"thresholds": [ "thresholds": [
0.1, 1.5,
0.3, 1.4,
0.5, 1.3,
0.6, 1.2,
0.7 1.1
], ],
"min_lesson_length": 100, "min_lesson_length": 100,
"signal_smoothing": true, "signal_smoothing": true,
......
...@@ -24,19 +24,6 @@ model_path = './models/{run_id}'.format(run_id=run_id) ...@@ -24,19 +24,6 @@ model_path = './models/{run_id}'.format(run_id=run_id)
summaries_dir = './summaries' summaries_dir = './summaries'
maybe_meta_curriculum = MetaCurriculum(curriculum_file) maybe_meta_curriculum = MetaCurriculum(curriculum_file)
def init_environment(env_path, docker_target_name, worker_id, seed):
if env_path is not None:
# Strip out executable extensions if passed
env_path = (env_path.strip()
.replace('.app', '')
.replace('.exe', '')
.replace('.x86_64', '')
.replace('.x86', ''))
docker_training = docker_target_name is not None
return
trainer_config = yaml.load(open(trainer_config_path)) trainer_config = yaml.load(open(trainer_config_path))
env = UnityEnvironment( env = UnityEnvironment(
n_arenas=1, # Change this to train on more arenas n_arenas=1, # Change this to train on more arenas
...@@ -49,7 +36,6 @@ env = UnityEnvironment( ...@@ -49,7 +36,6 @@ env = UnityEnvironment(
external_brains = {brain: env.brains[brain] for brain in env.external_brain_names} external_brains = {brain: env.brains[brain] for brain in env.external_brain_names}
# Create controller and begin training.
tc = TrainerController(model_path, summaries_dir, run_id + '-' + str(sub_id), tc = TrainerController(model_path, summaries_dir, run_id + '-' + str(sub_id),
save_freq, maybe_meta_curriculum, save_freq, maybe_meta_curriculum,
load_model, train_model, load_model, train_model,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment