首页 文章

Tensorflow摘要:添加一个不属于计算图的变量

提问于
浏览
8

我有一个变量随着列车迭代而变化 . 变量不作为计算图的一部分计算 .

是否可以将其添加到tensorflow摘要中,以便将其与损失函数一起显示?

2 回答

  • 15

    是的,您可以在图表外创建摘要 .

    以下是在图表外部创建摘要的示例(而不是TF op):

    output_path = "/tmp/myTest"
    summary_writer = tf.summary.FileWriter(output_path)
    
    for x in range(100):
       myVar = 2*x
    
       summary=tf.Summary()
       summary.value.add(tag='myVar', simple_value = myVar)
       summary_writer.add_summary(summary, x)
    
    summary_writer.flush()
    
  • 1

    如果您有其他摘要,则可以为变量添加新的占位符,而不是作为计算图的一部分计算的 .

    ...
    myVar_tf = tf.placeholder(dtype=tf.float32)
    tf.summary.scalar('myVar', myVar_tf)
    merged_summary = tf.summary.merge_all()
    ...
    ...
    myVar = 0.1
    feed_dict = { myVar_tf : myVar}
    summary, step = sess.run([merged_summary, global_step],feed_dict=feed_dict)
    summary_writer.add_summary(summary, step)
    

相关问题