Erwin Coumans committed 2017-10-20 20:32:49 -07:00
1 parent c480fab9a0
commit d18bfec1a9
3 files changed
+1 -5

No files matched your search

@@ -36,7 +36,7 @@ def default():
eval_episodes = 25 eval_episodes = 25
use_gpu = False use_gpu = False
# Network # Network
network = networks.ForwardGaussianPolicy network = networks.feed_forward_gaussian
weight_summaries = dict( weight_summaries = dict(
all=r'.*', all=r'.*',
policy=r'.*/policy/.*', policy=r'.*/policy/.*',
@@ -101,8 +101,6 @@ def train(config, env_processes):
""" """
tf.reset_default_graph() tf.reset_default_graph()
with config.unlocked: with config.unlocked:
config.network = functools.partial(
utility.define_network, config.network, config)
config.policy_optimizer = getattr(tf.train, config.policy_optimizer) config.policy_optimizer = getattr(tf.train, config.policy_optimizer)
config.value_optimizer = getattr(tf.train, config.value_optimizer) config.value_optimizer = getattr(tf.train, config.value_optimizer)
if config.update_every % config.num_agents: if config.update_every % config.num_agents:
@@ -98,8 +98,6 @@ def visualize(
""" """
config = utility.load_config(logdir) config = utility.load_config(logdir)
with config.unlocked: with config.unlocked:
config.network = functools.partial(
utility.define_network, config.network, config)
config.policy_optimizer = getattr(tf.train, config.policy_optimizer) config.policy_optimizer = getattr(tf.train, config.policy_optimizer)
config.value_optimizer = getattr(tf.train, config.value_optimizer) config.value_optimizer = getattr(tf.train, config.value_optimizer)
with tf.device('/cpu:0'): with tf.device('/cpu:0'):