Tensorboard to get gradients only for trainable weights
@@ -792,7 +792,7 @@ class TensorBoard(Callback):
if self.histogram_freq and self.merged is None:
for layer in self.model.layers:
- for weight in layer.weights:
+ for weight in layer.trainable_weights:
mapped_weight_name = weight.name.replace(':', '_')
tf.summary.histogram(mapped_weight_name, weight)
if self.write_grads:
