fix discrete versions of kuka grasping environment

should fix issue 1386
This commit is contained in:
Erwin Coumans committed 2017-10-31 16:14:20 -07:00
1 parent 33d32d7ab4
commit 3709e68645
3 files changed
+3 -3

No files matched your search

@@ -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:
@@ -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],
@@ -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,