hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[rllib] Forgot to register param noise layer variables

Browse source
This commit is contained in:
Eric Liang 2019-08-29 18:12:31 -07:00 • committed by GitHub
parent fe5bd09b46
commit 38231907f3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rllib/agents/ddpg/ddpg_model.py
View file

@ -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])