从之前的训练点在 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
。
我家乡面临严重停电,我不得不多次重新开始训练,关于如何从上一个迭代点恢复训练有什么建议吗? 我正在使用 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
。