pycaffe使用.solverstate文件继续训练

Posted houjun

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pycaffe使用.solverstate文件继续训练相关的知识,希望对你有一定的参考价值。

import caffe  
 
solver_file = "solver.prototxt"
solverstate = "xx.solverstate"
caffe.set_device(0)  
caffe.set_mode_gpu()
solver = caffe.get_solver(solver_file)
solver.restore(solverstate) 
solver.solve()

 

以上是关于pycaffe使用.solverstate文件继续训练的主要内容,如果未能解决你的问题,请参考以下文章

使用snapshot继续训练网络

Chapter 7 Windows下pycaffe的使用之draw_net.py

PyCaffe 构建失败,找不到 lboost_python

[caffe][原创]windows上pycaffe导入报错no module named _caffe

pycaffe简明文档

ubuntu配置caffe的python接口pycaffe