《深度学习——实战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()

如果是cpu，则改为mode_cpu()