fix typo, add train_ppo kuka task

2017-10-31 16:33:13 -07:00
parent 3709e68645
commit e4f58ddc33
3 changed files with 11 additions and 2 deletions
--- a/examples/pybullet/gym/pybullet_envs/agents/configs.py
+++ b/examples/pybullet/gym/pybullet_envs/agents/configs.py
@@ -100,6 +100,16 @@ def pybullet_ant():
  steps = 5e7  # 50M
  return locals()

+def pybullet_kuka_grasping():
+  """Configuration for Bullet Kuka grasping task."""
+  locals().update(default())
+  # Environment
+  env = 'KukaBulletEnv-v0'
+  max_length = 10
+  steps = 1e7  # 10M
+  return locals()
+
+
 def pybullet_racecar():
  """Configuration for Bullet MIT Racecar task."""
  locals().update(default())