如何使用 Tensorboard 检查训练模型的准确性？

Question

我正在运行用于图像分类的 CNN。 每 100 步，就会在文件夹中创建一个文件：model.ckpt-0.data-00000-of-00001、model.ckpt-0.index、model.ckpt-0.meta。 还有这些文件：graph.pbtxt 和 checkpoint。

我将使用这些文件中的哪个文件来查看 Tensorboard 中训练模型的准确性？

Answer 1

这些都不包含准确度值，它们是模型的定义（graph.pbtxt）和模型权重（checkpoint / ckpt 文件）。

默认情况下， fit方法将输出您在模型上调用compile时定义的任何损失或指标（例如准确性），例如

model.compile(optimizer="Adam", loss="mse", metrics=["mae", "acc"])

将使用mse损失以及mae和acc指标编译模型。 这些值将在每个时代结束时打印，或者如果您在调用fit时更改verbose参数，则会更频繁地打印

也许可视化这些值的最佳方法是使用Tensorboard 。 为此，您需要创建一个 tensorboard 回调（回调是一个具有在训练开始/结束时调用的方法、时期和批处理的类），它将指标和其他信息写入训练目录。

然后你可以从训练目录中运行 tensorboard，例如tensorboard --logdir=/path/to/training/dir以获得一个很好的基于 web 的 UI 来监控训练。