huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

120 episodes per rolloout

Browse Source
This commit is contained in:
ebuehrle
2022-02-28 22:41:39 +01:00
parent 261e35eae8
commit 7a134d98c9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
shail-experiment.py
View File

@@ -62,7 +62,7 @@ def training_function(config):
), options=option_list[config['policy']['option']], ), options=option_list[config['policy']['option']],
safe_actions_collision_method=config['trainenv']['safe_actions_collision_method'], safe_actions_collision_method=config['trainenv']['safe_actions_collision_method'],
abort_unsafe_collision_method=config['trainenv']['abort_unsafe_collision_method'], abort_unsafe_collision_method=config['trainenv']['abort_unsafe_collision_method'],
) for _ in range(60)] ) for _ in range(120)]
elif config['experiment'] == 'B': elif config['experiment'] == 'B':
envs = sum([[SafeOptionsEnv(Setobs( envs = sum([[SafeOptionsEnv(Setobs(
@@ -142,7 +142,7 @@ def training_function(config):
v_opt=v_opt, v_opt=v_opt,
v_iters=config['value']['iterations_per_epoch'], v_iters=config['value']['iterations_per_epoch'],
epochs=config['train_epochs'], epochs=config['train_epochs'],
rollout_episodes=60, rollout_episodes=120,
rollout_steps=60, rollout_steps=60,
gamma=0.99, gamma=0.99,
gae_lambda=0.9, gae_lambda=0.9,