caffe：如何在训练时保存最好的模型？答案

【问题标题】：caffe: How to save the best model while training?caffe：如何在训练时保存最好的模型？
【发布时间】：2017-03-28 08:44:19
【问题描述】：

我们在Caffe框架中训练一个模型时可能会测试多次，所以我想根据测试精度保存最好的模型，我该怎么办？谢谢！

【问题讨论】：

使快照迭代次数和测试迭代次数相同。
我想想，每个测试阶段我们保存模型，训练后我们选择最好的。但是我觉得不方便，所以我希望CAFFE只保存最好的。如果当前精度不高于以前的精度，我们不保存模型。我觉得应该修改源代码，但是我不清楚该怎么做。

标签： caffe

【解决方案1】：

您的求解器文件应包含诸如

之类的行

snapshot: 10000
snapshot_prefix: "models/bvlc_alexnet/caffe_alexnet_train"

第一个告诉 Caffe 多久检查一次（保存）模型参数和训练权重。第二个告诉存储它们的位置。请注意，您还将在训练结束时获得保存。文件显示为

caffe_alexnet_train_iter_50000.caffemodel
caffe_alexnet_train_iter_50000.solverstate

...对于在迭代 50,000 处保存的对。欢迎您重命名您认为合适的最佳集合，以便以后运行不会覆盖它们。我通常通过创建一个子目录 best 来做到这一点，并将最新的结果移到那里。

【讨论】：