用caffe训练模型时每隔一定的迭代次数保存一次caffemodel(snapshot)

利用ssd训练caffemode时,最好每隔一定的迭代次数就保存一个caffemodel及对应的solverstate,否则程序中途挂掉就得重新开始训练。

在example/ssd/中修改ssd_pascal.py文件中的snapshot值,将默认的80000改为500

从而使得在models/VGGNet/VOC0712/SSD_300×300/中修改solver.prototext中的snapshot一行

原文件中默认是80000,即迭代八万次才保存一个,改小一点,如500,1000。

猜你喜欢

转载自blog.csdn.net/qq_38469553/article/details/78190000
今日推荐