Allow variable horizon trajectories

2021-09-10 11:34:36 +02:00
parent b2b2abafa2
commit 544ea4d15a
3 changed files with 3 additions and 2 deletions
--- a/scratch/etienne/intersimple/gail_image.py
+++ b/scratch/etienne/intersimple/gail_image.py
@@ -47,6 +47,7 @@ gail_trainer = adversarial.GAIL(
    #n_disc_updates_per_round=2048,
    discrim_kwargs={'discrim_net': CnnDiscriminator(venv)},
    gen_algo=sb3.PPO("CnnPolicy", venv, verbose=1, n_steps=1024),
+    allow_variable_horizon=True,
 )
 gail_trainer.train(total_timesteps=100000)
 gail_trainer.gen_algo.save(model_name)