我正在运行tensorflow对象检测API,并使用SSD_mobilenet模型。我的训练目录中包含model.cpkt和graph.pbtxt。但是在我的培训目录中,我发现我的事件文件为空。似乎没有数据写入我的事件。有人可以帮我吗!
答案 0 :(得分:3)
Tensorflow事件文件将根据我们在代码中添加的摘要生成。
例如,假设您正在训练一个卷积神经网络来识别MNIST数字。您想记录学习率如何随时间变化以及目标函数如何变化。通过将tf.summary.scalar ops附加到分别输出学习率和损失的节点来收集这些信息。然后,给每个scalar_summary一个有意义的标签,例如“学习率”或“损失函数”。
例如:
tf.summary.scalar('损失',损失)
请参考以下链接: https://www.tensorflow.org/guide/summaries_and_tensorboard