TensorFlow2.0-for-Deep-Reinforcement-Learning icon indicating copy to clipboard operation
TensorFlow2.0-for-Deep-Reinforcement-Learning copied to clipboard

NoisyDense incorrect sigma init?

Open pedrohpf opened this issue 5 years ago • 0 comments

You have this code when init sigma:

sigma_initializer = tf.constant_initializer(self.std_init / np.sqrt(self.units))

Going into the original paper (section 3.2) I would assume the init to be like this:

sigma_initializer = tf.constant_initializer(self.std_init / np.sqrt(input_dim))

Was this change intended? Or maybe I'm misunderstanding the paper?

pedrohpf avatar Jan 16 '21 16:01 pedrohpf