diff options
Diffstat (limited to 'examples/pybullet/gym/pybullet_envs/minitaur/envs/minitaur_trotting_env_example.py')
-rw-r--r-- | examples/pybullet/gym/pybullet_envs/minitaur/envs/minitaur_trotting_env_example.py | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/examples/pybullet/gym/pybullet_envs/minitaur/envs/minitaur_trotting_env_example.py b/examples/pybullet/gym/pybullet_envs/minitaur/envs/minitaur_trotting_env_example.py index e158e1160..e38f86428 100644 --- a/examples/pybullet/gym/pybullet_envs/minitaur/envs/minitaur_trotting_env_example.py +++ b/examples/pybullet/gym/pybullet_envs/minitaur/envs/minitaur_trotting_env_example.py @@ -26,13 +26,12 @@ def main(argv): network = config.network with tf.Session() as sess: - agent = simple_ppo_agent.SimplePPOPolicy( - sess, - env, - network, - policy_layers=policy_layers, - value_layers=value_layers, - checkpoint=os.path.join(LOG_DIR, CHECKPOINT)) + agent = simple_ppo_agent.SimplePPOPolicy(sess, + env, + network, + policy_layers=policy_layers, + value_layers=value_layers, + checkpoint=os.path.join(LOG_DIR, CHECKPOINT)) sum_reward = 0 observation = env.reset() @@ -48,4 +47,3 @@ def main(argv): if __name__ == "__main__": tf.app.run(main) - |