|
点击此处返回总目录
如果一个模型很大，训练到一般，突然停电了。这时候就要从头再来一遍了。 snapshot技术就是在训练过程中，把参数记录下来。以后可以从保存记录的时刻开始继续训练。
我们在训练的时候，会在超参数文件中设置没多少轮保存一份模型。比如，希望总共训练1万次，设置了每5000次保存一份模型文件。
当我们训练了6000次之后，停电了！
这时候，我们在models文件夹下生成了 其中，第一个文件是caffe的模型文件。第二个文件会记录整个运行过程，运行到哪里了，参数是多少。我们一会就调用第二个文件，去继续训练。
我们想下次训练的时候从5000次继续训练，具体怎么做呢？ 在训练的语句中加入一个参数即可。
运行，可以看到，是接着5000次继续训练。
|