caffe:如何在训练时保存最好的模型?

时间:2017-03-28 08:44:19

标签: caffe

我们可以在框架Caffe中训练模型时多次测试,所以我想根据测试精度保存最好的模型,那么我该怎么办?谢谢!

1 个答案:

答案 0 :(得分:2)

您的求解器文件应包含

等行
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中保存的对。您可以根据需要重命名最佳设置,因此以后的运行不会覆盖它们。我通常通过创建一个子目录最佳来做到这一点,并在那里移动最新的结果。