From d18bfec1a99e7e82acd32da3324487768393fdae Mon Sep 17 00:00:00 2001 From: Erwin Coumans Date: Fri, 20 Oct 2017 20:32:49 -0700 Subject: [PATCH] API changes in TensorFlow Agents See https://github.com/tensorflow/agents/commit/164f62032696e06f616268a51d32ca6b84dcbf15#diff-82bda908f176861c20aadc1d017c2527 --- examples/pybullet/gym/pybullet_envs/agents/configs.py | 2 +- examples/pybullet/gym/pybullet_envs/agents/train_ppo.py | 2 -- examples/pybullet/gym/pybullet_envs/agents/visualize_ppo.py | 2 -- 3 files changed, 1 insertion(+), 5 deletions(-) diff --git a/examples/pybullet/gym/pybullet_envs/agents/configs.py b/examples/pybullet/gym/pybullet_envs/agents/configs.py index 9b2cb2da2..6e787d129 100644 --- a/examples/pybullet/gym/pybullet_envs/agents/configs.py +++ b/examples/pybullet/gym/pybullet_envs/agents/configs.py @@ -36,7 +36,7 @@ def default(): eval_episodes = 25 use_gpu = False # Network - network = networks.ForwardGaussianPolicy + network = networks.feed_forward_gaussian weight_summaries = dict( all=r'.*', policy=r'.*/policy/.*', diff --git a/examples/pybullet/gym/pybullet_envs/agents/train_ppo.py b/examples/pybullet/gym/pybullet_envs/agents/train_ppo.py index 6b6a49766..561de7628 100644 --- a/examples/pybullet/gym/pybullet_envs/agents/train_ppo.py +++ b/examples/pybullet/gym/pybullet_envs/agents/train_ppo.py @@ -101,8 +101,6 @@ def train(config, env_processes): """ tf.reset_default_graph() with config.unlocked: - config.network = functools.partial( - utility.define_network, config.network, config) config.policy_optimizer = getattr(tf.train, config.policy_optimizer) config.value_optimizer = getattr(tf.train, config.value_optimizer) if config.update_every % config.num_agents: diff --git a/examples/pybullet/gym/pybullet_envs/agents/visualize_ppo.py b/examples/pybullet/gym/pybullet_envs/agents/visualize_ppo.py index f6673dc1d..f7a67c7ea 100644 --- a/examples/pybullet/gym/pybullet_envs/agents/visualize_ppo.py +++ b/examples/pybullet/gym/pybullet_envs/agents/visualize_ppo.py @@ -98,8 +98,6 @@ def visualize( """ config = utility.load_config(logdir) with config.unlocked: - config.network = functools.partial( - utility.define_network, config.network, config) config.policy_optimizer = getattr(tf.train, config.policy_optimizer) config.value_optimizer = getattr(tf.train, config.value_optimizer) with tf.device('/cpu:0'):