1

我指的是Tensorflow提供的用于图像分类的train_image_classifier.py脚本。

脚本使用 slim.learning.train API 来启动训练,然后它会在内部处理所有事情。我的要求是在每 n 个 epoch(即 5 个)之后提取损失和准确度指标,并将其转储到数据库中以供以后分析。

但是由于 slim train API 的高级抽象,它不会返回损失/准确性信息。

有没有办法在给定数量的步骤/时期后利用训练过程并获得损失/准确度值?如果可能的话,请分享它的伪代码。

这对我真的很有帮助。

谢谢。

4

0 回答 0