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.cnblogs.com/denny402/p/5679204.html

相关资源：各显卡算力对照表！