如果不进行可视化,只想得到一个最终的训练model, 那么代码非常简单,如下 :
import caffe caffe.set_device(0) caffe.set_mode_gpu() solver = caffe.SGDSolver('/home/xxx/data/solver.prototxt') solver.solve()