我们可以在框架Caffe中训练模型时多次测试,所以我想根据测试精度保存最好的模型,那么我该怎么办?谢谢!
答案 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中保存的对。您可以根据需要重命名最佳设置,因此以后的运行不会覆盖它们。我通常通过创建一个子目录最佳来做到这一点,并在那里移动最新的结果。