首页 > 编程 > Python > 正文

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

2019-11-06 08:25:11
字体:
来源:转载
供稿:网友

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

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

 


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表