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()

 

posted @   denny402  阅读(22944)  评论(3编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示