Port TRPO, PPO, GAIL

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

View File

@@ -0,0 +1,8 @@
from sb3_contrib import TRPO
import gym
from gym.wrappers import TransformObservation
env = TransformObservation(gym.make('Pendulum-v0'), lambda obs: obs)
model = TRPO("MlpPolicy", env, verbose=1)
model.learn(total_timesteps=250000)