From 38231907f348a8a5521fed0cf2613658a349601b Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Thu, 29 Aug 2019 18:12:31 -0700 Subject: [PATCH] [rllib] Forgot to register param noise layer variables --- rllib/agents/ddpg/ddpg_model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rllib/agents/ddpg/ddpg_model.py b/rllib/agents/ddpg/ddpg_model.py index 8fb7c4a9a..f71079640 100644 --- a/rllib/agents/ddpg/ddpg_model.py +++ b/rllib/agents/ddpg/ddpg_model.py @@ -258,3 +258,5 @@ class DDPGModel(TFModelV2): add_noise_ops.append(tf.assign_add(var, var_noise)) self.add_noise_op = tf.group(*tuple(add_noise_ops)) self.pi_distance = None + self.register_variables(self.parameter_noise) + self.register_variables([self.parameter_noise_sigma])