We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1f04c1d commit bc7ee05Copy full SHA for bc7ee05
PolicyGradient/a3c/worker.py
@@ -85,7 +85,7 @@ def __init__(self, name, env, policy_net, value_net, global_counter, discount_fa
85
# Op to copy params from global policy/valuenets
86
self.copy_params_op = make_copy_params_op(
87
tf.contrib.slim.get_variables(scope="global", collection=tf.GraphKeys.TRAINABLE_VARIABLES),
88
- tf.contrib.slim.get_variables(scope=self.name, collection=tf.GraphKeys.TRAINABLE_VARIABLES))
+ tf.contrib.slim.get_variables(scope=self.name+'/', collection=tf.GraphKeys.TRAINABLE_VARIABLES))
89
90
self.vnet_train_op = make_train_op(self.value_net, self.global_value_net)
91
self.pnet_train_op = make_train_op(self.policy_net, self.global_policy_net)
0 commit comments