1. 程式人生 > >用caffe訓練模型時每隔一定的迭代次數儲存一次caffemodel(snapshot)

用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。