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