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

Resume training in Caffe from the previous training point

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

Caffe 每隔一段时间可以保存一个"snapshot"。您可以简单地从上次快照恢复训练:

$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