Snapshot使用

                                                                                                                                                点击此处返回总目录

 

如果一个模型很大,训练到一般,突然停电了。这时候就要从头再来一遍了。

snapshot技术就是在训练过程中,把参数记录下来。以后可以从保存记录的时刻开始继续训练。

 

 

我们在训练的时候,会在超参数文件中设置没多少轮保存一份模型。比如,希望总共训练1万次,设置了每5000次保存一份模型文件。

    Snapshot使用_第1张图片

 

当我们训练了6000次之后,停电了!

 

这时候,我们在models文件夹下生成了

                

其中,第一个文件是caffe的模型文件。第二个文件会记录整个运行过程,运行到哪里了,参数是多少。我们一会就调用第二个文件,去继续训练。

 

我们想下次训练的时候从5000次继续训练,具体怎么做呢?

在训练的语句中加入一个参数即可。

 

 

运行,可以看到,是接着5000次继续训练。

          Snapshot使用_第2张图片

 

 

 

 

 

 

你可能感兴趣的:(17,深度学习)