Bart/bullet3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix discrete versions of kuka grasping environment

Browse Source 
should fix issue 1386
This commit is contained in:
Erwin Coumans
2017-10-31 16:14:20 -07:00
parent 33d32d7ab4
commit 3709e68645
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pybullet/gym/pybullet_envs/baselines/enjoy_kuka_grasping.py
View File

@@ -12,7 +12,7 @@ from baselines import deepq
def main(): def main():
env = KukaGymEnv(renders=True) env = KukaGymEnv(renders=True, isDiscrete=True)
act = deepq.load("kuka_model.pkl") act = deepq.load("kuka_model.pkl")
print(act) print(act)
while True: while True:

2
examples/pybullet/gym/pybullet_envs/baselines/train_kuka_cam_grasping.py
View File

@@ -25,7 +25,7 @@ def callback(lcl, glb):
def main(): def main():
env = KukaCamGymEnv(renders=True) env = KukaCamGymEnv(renders=False, isDiscrete=True)
model = deepq.models.cnn_to_mlp( model = deepq.models.cnn_to_mlp(
convs=[(32, 8, 4), (64, 4, 2), (64, 3, 1)], convs=[(32, 8, 4), (64, 4, 2), (64, 3, 1)],
hiddens=[256], hiddens=[256],

2
examples/pybullet/gym/pybullet_envs/baselines/train_kuka_grasping.py
View File

@@ -25,7 +25,7 @@ def callback(lcl, glb):
def main(): def main():
env = KukaGymEnv(renders=False) env = KukaGymEnv(renders=False, isDiscrete=True)
model = deepq.models.mlp([64]) model = deepq.models.mlp([64])
act = deepq.learn( act = deepq.learn(
env, env,