From 7a134d98c9cd1c4735f662d45eae573d1eef3fd5 Mon Sep 17 00:00:00 2001 From: ebuehrle <43623224+ebuehrle@users.noreply.github.com> Date: Mon, 28 Feb 2022 22:41:39 +0100 Subject: [PATCH] 120 episodes per rolloout --- shail-experiment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/shail-experiment.py b/shail-experiment.py index f00699d..9964441 100644 --- a/shail-experiment.py +++ b/shail-experiment.py @@ -62,7 +62,7 @@ def training_function(config): ), options=option_list[config['policy']['option']], safe_actions_collision_method=config['trainenv']['safe_actions_collision_method'], abort_unsafe_collision_method=config['trainenv']['abort_unsafe_collision_method'], - ) for _ in range(60)] + ) for _ in range(120)] elif config['experiment'] == 'B': envs = sum([[SafeOptionsEnv(Setobs( @@ -142,7 +142,7 @@ def training_function(config): v_opt=v_opt, v_iters=config['value']['iterations_per_epoch'], epochs=config['train_epochs'], - rollout_episodes=60, + rollout_episodes=120, rollout_steps=60, gamma=0.99, gae_lambda=0.9,