码迷,mamicode.com
首页 > 其他好文 > 详细

tensorflow summary

时间:2018-08-28 18:58:35      阅读:214      评论:0      收藏:0      [点我收藏+]

标签:word   attention   run   session   rap   color   cal   log   style   

定义summary

writer = tf.summary.FileWriter(logdir=self.han_config.log_path, graph=session.graph)

 

1.scalar存储结果

  a.先在训练的循环外定义:

test_accuracy_summary = tf.summary.scalar(test_accuracy, self.han_model.accuracy)
            test_loss_summary = tf.summary.scalar(test_loss, self.han_model.loss)
            test_scalar = tf.summary.merge([test_accuracy_summary, test_loss_summary])

 

  b.在session run的时候run test_scalar,获得值,然后再添加。

 writer.add_summary(summary=train_scalar_, global_step=steps)

 

 

2.histogram存储权重,偏执。

  a.先在训练的循环外定义:

            W_w_attention_word_histogram = tf.summary.histogram(W_w_attention_word, self.han_model.W_w_attention_word)
            W_b_attention_word_histogram = tf.summary.histogram(W_w_attention_word, self.han_model.W_b_attention_word)
            context_vecotor_word_histogram = tf.summary.histogram(context_vecotor_word,
                                                                  self.han_model.context_vecotor_word)
            W_w_attention_sentence_histogram = tf.summary.histogram(W_w_attention_sentence,
                                                                    self.han_model.W_w_attention_sentence)
            W_b_attention_sentence_histogram = tf.summary.histogram(W_b_attention_sentence,
                                                                    self.han_model.W_b_attention_sentence)
            context_vecotor_sentence_histogram = tf.summary.histogram(context_vecotor_sentence,
                                                                      self.han_model.context_vecotor_sentence)
            train_variable_histogram = tf.summary.merge([W_w_attention_word_histogram, W_b_attention_word_histogram,
                                                    context_vecotor_word_histogram, W_w_attention_sentence_histogram,
                                                W_b_attention_sentence_histogram, context_vecotor_sentence_histogram])

 

  b.在session run的时候run test_scalar,获得值,然后再添加。

writer.add_summary(summary=train_variable_histogram_, global_step=steps)

 

tensorflow summary

标签:word   attention   run   session   rap   color   cal   log   style   

原文地址:https://www.cnblogs.com/callyblog/p/9549993.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!