caffe的python接口学习训练模型training
Posted 牛郎
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了caffe的python接口学习训练模型training相关的知识，希望对你有一定的参考价值。
如果不进行可视化，只想得到一个最终的训练model, 那么代码非常简单，如下 :
import caffe
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.SGDSolver(‘/home/xxx/data/solver.prototxt‘)
solver.solve()
以上是关于caffe的python接口学习训练模型training的主要内容，如果未能解决你的问题，请参考以下文章
Caffe-python interface 学习|网络训练部署测试
Caffe-python interface 学习|网络训练部署測试