当前位置: 代码迷 >> 综合 >> 用caffe训练模型时每隔一定的迭代次数保存一次caffemodel(snapshot)
  详细解决方案

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

热度:73   发布时间:2023-12-21 04:23:44.0

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

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

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

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

  相关解决方案