caffe的python接口学习(3):训练模型(training)

如果不进行可视化,只想得到一个最终的训练model, 那么代码非常简单,如下 :

import caffe caffe.set_device(0) caffe.set_mode_gpu() solver = caffe.SGDSolver(\'/home/xxx/data/solver.prototxt\') solver.solve()

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/zgzdss.html