1. 程式人生 > >【5】caffe的python介面學習:訓練模型(training)

【5】caffe的python介面學習:訓練模型(training)

如果不進行視覺化,只想得到一個最終的訓練model, 那麼程式碼非常簡單,如下 :

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