pycaffe网络模型训练和fine-tuning

GPU训练：

import caffe

caffe.set_mode_gpu()
caffe.set_device(1)    #1块显卡

solver_file= '/root/Files/solver.prototxt'

solver = caffe.get_solver(solver_file)
solver.solve()

GPU fine-tuning：

import caffe

caffe.set_mode_gpu()
caffe.set_device(1)    #1块显卡

solver_file = '/yourPath/solver.prototxt'
weights = '/yourPath/inter_1200000.caffemodel'

solver = caffe.get_solver(solver_file)
solver.net.copy_from(weights)
solver.solve()

训练中断，继续训练GPU:

import caffe

caffe.set_mode_gpu()
caffe.set_device(1)    #1块显卡

solver_file = '/yourPath/solver.prototxt'
solverstate = "/yourPath/_iter_5000.solverstate"

solver = caffe.get_solver(solver_file)
solver.restore(solverstate) #就用这一句
solver.solve()

 

CPU训练fine-tuning：

caffe.set_mode_cpu()