【发布时间】:2017-08-13 07:44:16
【问题描述】:
我的家乡面临严重的停电,我不得不多次重新开始训练,有什么建议可以从上一个迭代点恢复训练吗? 我正在使用 caffe 和 lmdb 文件。 提前致谢
【问题讨论】:
标签: caffe
我的家乡面临严重的停电,我不得不多次重新开始训练,有什么建议可以从上一个迭代点恢复训练吗? 我正在使用 caffe 和 lmdb 文件。 提前致谢
【问题讨论】:
标签: caffe
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。
【讨论】: