Allow variable horizon trajectories

2021-09-10 11:34:36 +02:00
parent b2b2abafa2
commit 9a107b165a
2 changed files with 2 additions and 0 deletions
--- a/scratch/etienne/intersimple/gail_flat.py
+++ b/scratch/etienne/intersimple/gail_flat.py
@@ -47,6 +47,7 @@ gail_trainer = adversarial.GAIL(
    n_disc_updates_per_round=32,
    discrim_kwargs={'discrim_net': MlpDiscriminator()},
    gen_algo=sb3.PPO("MlpPolicy", venv, verbose=1, n_steps=4530),
+    allow_variable_horizon=True,
 )
 gail_trainer.train(total_timesteps=400000)
 gail_trainer.gen_algo.save(model_name)