tensorflow在训练和验证时监视不同的summary的操作
生活随笔
收集整理的這篇文章主要介紹了
tensorflow在训练和验证时监视不同的summary的操作
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
如果想在訓練和驗證時監視不同的summary,將train summary ops和val summary ops放進不同的集合中即可。
train_writer = tf.summary.FileWriter(log_dir + '/train', sess.graph)
val_writer = tf.summary.FileWriter(log_dir + '/val', sess.graph)# 假設train_loss和val_loss的計算方式不同
tf.summary.scalar("train_loss", train_loss, collections=['train'])
tf.summary.scalar("val_loss", val_loss, collections=['val'])train_summary_ops = tf.summary.merge_all('train')
val_summary_ops = tf.summary.merge_all('val')# training
...
train_summary = sess.run(train_summary_ops, feed_dict=train_dict)
train_writer.add_summary(train_summary, global_step)# validation
...
val_summary = sess.run(val_summary_ops, feed_dict=val_dict)
val_writer.add_summary(val_summary, global_step)# end
train_writer.close()
val_writer.close()
轉載于:https://www.cnblogs.com/MrZJ/p/9831543.html
總結
以上是生活随笔為你收集整理的tensorflow在训练和验证时监视不同的summary的操作的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 16年古井原浆多少钱一瓶?
- 下一篇: (原)ubuntu中使用conda安装t