[rllib] Forgot to register param noise layer variables

2026-07-29 11:26:04 +08:00 · 2019-08-29 18:12:31 -07:00
parent fe5bd09b46
commit 38231907f3
1 changed files with 2 additions and 0 deletions
@@ -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])