从之前的训练点恢复Caffe的训练

Question

我在家乡面临严重停电，我不得不多次重新开始我的训练，有关如何从上一次迭代点恢复训练的任何建议？ 我正在使用caffe和lmdb文件。 提前致谢

Answer 1

Caffe可以偶尔保存一个“快照”。您可以通过简单的方式从上一张快照恢复训练：

$CAFFE_ROOT/build/tools/caffe train -model /path/to/solver.prototxt -snapshot /path/to/latest.solverstate

在solver.prototxt中，您可以通过设置

来定义拍摄快照的频率

snapshot: 2500  # take a snapshot every 2500 iterations

快照文件保存到

定义的相同位置

snapshot_prefix: "/path/to/snaps"

在那里，您可以找到为每2500次迭代保存的.solverstate和.caffemodel。