使用 Caffe 训练时如何获取训练错误？

Question

我正在使用 Caffe 训练一个简单的神经网络，来自 MNIST 示例。训练时我需要做两件事：

1. 每 600 次迭代训练集上的分类误差
2. 经过 30000 次迭代，最后测试集上的分类错误

这可能吗？

我发现了如何做 1) 通过用训练集替换测试集并使用test_iter，但它似乎真的很糟糕，然后我不能拥有 2)。例如，有没有一种简单的方法来告诉 Caffe 计算每次 display 迭代的训练误差？

Answer 1

1. 也许您可以在.prototxt 文件中添加一个测试阶段并使用 Accuracy layer 输出准确率。
2. 使用test_interval: 600 设置测试间隔。