denny402

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

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

 

分类:

技术点:

相关文章: