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

来源:互联网 发布:wps设置数据有效性 编辑:程序博客网 时间:2024/04/30 18:33

转载原文地址:http://www.cnblogs.com/denny402/p/5679204.html

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

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

 

0 0
原创粉丝点击