您的位置：首页 > 编程语言 > Python开发

caffe的python接口学习（3）：训练模型（training)

2017-01-07 11:09 363 查看
原文链接：http://www.cnblogs.com/denny402/p/5679204.html

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

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


 
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签： 
相关文章推荐
新的分享
章节导航