Tf Summary не дает гистограммы, но сохраняет график сеанса (предварительно обученная модель) - PullRequest
Новая
       43

Tf Summary не дает гистограммы, но сохраняет график сеанса (предварительно обученная модель)

1 голос
/ 20 марта 2019

Я выполняю анализ, чтобы визуализировать распределение весов для предварительно обученной модели, доступной онлайн.Это модель Resnet18, обученная на CIFAR10.

У меня есть следующий код для восстановления модели из meta и ckpt, а затем я пытаюсь создать гистограмму всех weights и bias слоев свертки, используя tf.summary.histogram

`with tf.Session(graph=tf.Graph()) as sess:
            read=tf.train.import_meta_graph(self.paths[0], clear_devices=True)
            try:
                read.restore(sess, tf.train.latest_checkpoint(self.paths[1]))
            except ValueError:
                try:
                    read.restore(sess, self.paths[1])
                except Exception as e:
                    print(e.message)

            # Summaries of weights
            summ_writer = tf.summary.FileWriter(self.sum_path, sess.graph)
            fp_summaries = []
            for lys in tf.trainable_variables():
                lay_nam = lys.name.split("/")[-2]
                if 'kernel' in lys.name:
                    with tf.name_scope(lay_nam+'_hist'):
                        tf_w_hist = tf.summary.histogram('Weights', tf.reshape(lys.eval(), [-1]))
                        fp_summaries.extend([tf_w_hist])
                if 'bias' in lys.name:
                    with tf.name_scope(lay_nam+'_hist'):
                        tf_b_hist = tf.summary.histogram('Bias', lys.eval())
                        fp_summaries.extend([tf_b_hist])
            tf_fp_summaries = tf.summary.merge(fp_summaries)
            # Run the graph
            output, _=sess.run([softmax, tf_fp_summaries], feed_dict={x: self.x_test[0:100, ]})

Однако в журнале событий, хранящихся в папке, хранится только основной график.Гистограммы не видны на tensorboard.Что может быть не так?

1 Ответ

1 голос
/ 20 марта 2019

Недостаточно передать объединенный сводный узел в sess.run. Вам нужно взять этот оцененный результат и передать его add_summary методу вашего FileWriter экземпляра. 

# evaluate the merged summary node in the graph
output, summ = sess.run([softmax, tf_fp_summaries], ...)
# explicitly write to file
summ_writer.add_summary(summ, global_step)
# optional, force to write to disk
summ_writer.flush()
...