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文件继续训练的主要内容,如果未能解决你的问题,请参考以下文章
Chapter 7 Windows下pycaffe的使用之draw_net.py
PyCaffe 构建失败,找不到 lboost_python