add a few more discrete actions to cartpole_bullet.py so it trains faster + add option to train without rendering, enjoy with rendering.

2017-06-22 16:49:14 -07:00
parent fbff4c57ca
commit a136098120
2 changed files with 11 additions and 7 deletions
--- a/examples/pybullet/gym/train_pybullet_cartpole.py
+++ b/examples/pybullet/gym/train_pybullet_cartpole.py
@@ -12,7 +12,7 @@ def callback(lcl, glb):

 def main():
 	
-    env = gym.make('CartPoleBulletEnv-v0')
+    env = CartPoleBulletEnv(renders=False)
    model = deepq.models.mlp([64])
    act = deepq.learn(
        env,