在进行模型训练时,需要同时将训练loss曲线和测试loss曲线显示在一个曲线图中,便于观察网络训练情况(是否过拟合、欠拟合)。
train_writer = tf.summary.FileWriter(FLAGS.log_dir+'train', sess.graph)
eval_writer = tf.summary.FileWriter(FLAGS.log_dir+'eval') #注意不需要加 sess.graph
# train_summary
train_writer.add_summary(train_summary, global_step)
train_writer.flush()
# eval_summary
eval_writer.add_summary(eval_summary, global_step)
eval_writer.flush()