Port TRPO, PPO, GAIL
This commit is contained in:
8
scratch/etienne/trpo/sb3/sb3-trpo-pendulum.py
Normal file
8
scratch/etienne/trpo/sb3/sb3-trpo-pendulum.py
Normal 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)
|
||||
Reference in New Issue
Block a user