def grad_histograms(grads_and_vars):
s = []
for grad, var in grads_and_vars:
s.append(tf.histogram_summary(var.op.name + '', var))
s.append(tf.histogram_summary(var.op.name + '/gradients', grad))
return tf.merge_summary(s)
评论列表
文章目录