tensorboard可视化训练之train_loss和eval_loss放在同一曲线图中

在进行模型训练时,需要同时将训练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()

效果:

tensorboard可视化训练之train_loss和eval_loss放在同一曲线图中_第1张图片

你可能感兴趣的:(实战总结)