From 5c3bf5dbf36d66e75c828e9060cb72002d88f064 Mon Sep 17 00:00:00 2001 From: Dustin Tran Date: Thu, 20 Dec 2018 11:56:39 -0800 Subject: [PATCH] Change string manipulation for gradient summaries. --- edward/inferences/variational_inference.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/edward/inferences/variational_inference.py b/edward/inferences/variational_inference.py index f6e8c244d..315cfedbc 100644 --- a/edward/inferences/variational_inference.py +++ b/edward/inferences/variational_inference.py @@ -71,11 +71,10 @@ def initialize(self, optimizer=None, var_list=None, use_prettytensor=False, tf.summary.scalar("loss", self.loss, collections=[self._summary_key]) for grad, var in grads_and_vars: # replace colons which are an invalid character - tf.summary.histogram("gradient/" + - var.name.replace(':', '/'), + var_name = var.name.replace(':', '/') + tf.summary.histogram("gradient/{}".format(var_name), grad, collections=[self._summary_key]) - tf.summary.scalar("gradient_norm/" + - var.name.replace(':', '/'), + tf.summary.scalar("gradient_norm/{}".format(var_name), tf.norm(grad), collections=[self._summary_key]) self.summarize = tf.summary.merge_all(key=self._summary_key)