如果我们在训练途中停电或者有了其他的情况,可以通过之前保存的状态恢复数据,使用的时候直接添加–snapshot参数即可,如:
./build/tools/caffe train --solver=models/bvlc_reference_caffenet/solver.prototxt --snapshot=models/bvlc_reference_caffenet/caffenet_train_iter_10000.solverstate
这时候运行会从snapshot开始继续运行,如从第600迭代时运行: