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的操作相关推荐

keras用cpu加速_在训练某些网络时，Keras（Tensorflow后端）在GPU上比在CPU上慢CPU
我很难理解GPU和CPU速度与小尺寸网络(CPU有时更快)相似的原因,而GPU更快更大尺寸的网络.问题底部的代码在i7-6700k上以103.7s运行,但是当使用tensorflow-gpu时,代码运 ...
TensorFlow estimator训练时 invalid continuation byte
** TensorFlow estimator训练时报错:invalid continuation byte ** 报错:UnicodeDecodeError: 'utf-8' codec can' ...
tensorflow yolov3训练自己的数据集，详细教程
这个教程是我在自己学习的过程中写的,当作一个笔记,写的比较详细在github上下载yolov3的tensorflow1.0版本: https://github.com/YunYang1994/ten ...
图像语义分割实战：TensorFlow Deeplabv3+ 训练自己数据集
文章目录前言一.环境配置二.训练过程 1.引入库 2.数据集准备转换为 VOC 格式的数据集 Convert to 灰度图 Convert to tfrecord 3.训练前代码准备 4.主要 ...
如何在TensorFlow中训练Boosted Trees模型
在使用结构化数据时,诸如梯度提升决策树和随机森林之类的树集合方法是最流行和最有效的机器学习工具之一. 树集合方法训练速度快,无需大量调整即可正常工作,并且不需要大型数据集进行训练. 在TensorFl ...
十折交叉验证10-fold cross validation, 数据集划分训练集验证集测试集
机器学习数据挖掘数据集划分训练集验证集测试集 Q:如何将数据集划分为测试数据集和训练数据集? A:three ways: 1.像sklearn一样,提供一个将数据集切分成训练集和测试集的函数 ...
机器学习中的训练集验证集测试集的关系
1.划分测试集目的为了了解一个模型对新样本的泛化能力,唯一的办法是:让已经训练好的模型真正的处理新的样本. 解决方法: 将原始数据划分成两个部分:训练集测试集.可以使用训练集来训练模型,然后用测试 ...
深度学习总结：用pytorch做dropout和Batch Normalization时需要注意的地方，用tensorflow做dropout和BN时需要注意的地方,
用pytorch做dropout和BN时需要注意的地方 pytorch做dropout: 就是train的时候使用dropout,训练的时候不使用dropout, pytorch里面是通过net.ev ...
训练集验证集_训练与验证、测试集数据分布不同的情况
在不同分布的数据集上进行训练与验证.测试深度学习需要大量的数据,但是有时我们可获得的满足我们真实需求分布的数据并不是那么多,不足以对我们的模型进行训练.这时我们就会收集大量相关的数据加入到训练集中, ...

tensorflow在训练和验证时监视不同的summary的操作

tensorflow在训练和验证时监视不同的summary的操作相关推荐

最新文章

热门文章