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

remove sonnet dependency

Browse Source
This commit is contained in:
Jie Tan
2017-05-18 16:12:38 -07:00
parent d40787f3fd
commit 8a6a46d180
11 changed files with 41 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/pybullet/gym/minitaurGymEnvTest.py
View File

@@ -10,7 +10,7 @@ import numpy as np
import tensorflow as tf
from envs.bullet.minitaurGymEnv import MinitaurGymEnv
from agents import simpleAgent
from agents import simplerAgent
def testSinePolicy():
"""Tests sine policy
@@ -53,17 +53,14 @@ def testDDPGPolicy():
environment = MinitaurGymEnv(render=True)
sum_reward = 0
steps = 1000
ckpt_path = 'data/agent/tf_graph_data/tf_graph_data.ckpt'
ckpt_path = 'data/agent/tf_graph_data/tf_graph_data_converted.ckpt-0'
observation_shape = (31,)
action_size = 8
actor_layer_sizes = (100, 181)
n_steps = 0
tf.reset_default_graph()
with tf.Session() as session:
agent = simpleAgent.SimpleAgent(session, ckpt_path,
actor_layer_sizes,
observation_size=observation_shape,
action_size=action_size)
agent = simplerAgent.SimplerAgent(session, ckpt_path)
state = environment.reset()
action = agent(state)
for _ in range(steps):