修改代码崩溃一上午,找了一些文章加上与源代码相比发现是summaries的语句顺序问题
首先应该是
tf.summary.scalar()
这句是给需要画出来的函数一个名字,相当于初始化
tf.summary.merge_all()
这表示将所有summary全部保存到磁盘,以便tensorboard显示
之后经过with tf.Session() as sess:开始准备训练
然后就是加入保存位置然后准备写入
os.path.join()
tf.summary.FileWriter()
最后就是将数据进行保存
if is_training:
train_summary_writer.add_summary(summaries, step)
else:
valid_summary_writer.add_summary(summaries, step)
总结起来就是
loss_summary = tf.summary.scalar('Loss', model.loss)
train_summary_op= tf.summary.merge_all()
train_summary_dir = os.path.join(outdir, 'summaries', 'train')
train_summary_writer = tf.summary.FileWriter(train_summary_dir, sess.graph)
for step in xrange(training_step):
summaries= sess.run(train_summary_op,feed_dic