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

Port TRPO, PPO, GAIL

Browse Source
This commit is contained in:
ebuehrle
2022-02-15 11:01:52 +01:00
parent 530ac95d61
commit a3b9b3e250
79 changed files with 5633 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scratch/etienne/trpo/sb3/sb3-ppo-pendulum.py Normal file
View File

@@ -0,0 +1,6 @@
from stable_baselines3 import PPO
from stable_baselines3.common.env_util import make_vec_env
env = make_vec_env("Pendulum-v0", n_envs=4)
model = PPO("MlpPolicy", env, verbose=1)
model.learn(total_timesteps=250000)