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

Allow variable horizon trajectories

Browse Source
This commit is contained in:
Johannes Fischer
2021-09-10 11:34:36 +02:00
parent b2b2abafa2
commit 9a107b165a
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
scratch/etienne/intersimple/gail_flat.py
View File

@@ -47,6 +47,7 @@ gail_trainer = adversarial.GAIL(
n_disc_updates_per_round=32, n_disc_updates_per_round=32,
discrim_kwargs={'discrim_net': MlpDiscriminator()}, discrim_kwargs={'discrim_net': MlpDiscriminator()},
gen_algo=sb3.PPO("MlpPolicy", venv, verbose=1, n_steps=4530), gen_algo=sb3.PPO("MlpPolicy", venv, verbose=1, n_steps=4530),
allow_variable_horizon=True,
) )
gail_trainer.train(total_timesteps=400000) gail_trainer.train(total_timesteps=400000)
gail_trainer.gen_algo.save(model_name) gail_trainer.gen_algo.save(model_name)

1
scratch/etienne/intersimple/gail_image.py
View File

@@ -47,6 +47,7 @@ gail_trainer = adversarial.GAIL(
#n_disc_updates_per_round=2048, #n_disc_updates_per_round=2048,
discrim_kwargs={'discrim_net': CnnDiscriminator(venv)}, discrim_kwargs={'discrim_net': CnnDiscriminator(venv)},
gen_algo=sb3.PPO("CnnPolicy", venv, verbose=1, n_steps=1024), gen_algo=sb3.PPO("CnnPolicy", venv, verbose=1, n_steps=1024),
allow_variable_horizon=True,
) )
gail_trainer.train(total_timesteps=100000) gail_trainer.train(total_timesteps=100000)
gail_trainer.gen_algo.save(model_name) gail_trainer.gen_algo.save(model_name)