参考:点击打开链接
import caffe
solver_file = "solver.prototxt"
solverstate = "models-12/_iter_5000.solverstate"
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.get_solver(solver_file)
solver.restore(solverstate) #就用这一句
solver.solve()
参考:点击打开链接
import caffe
solver_file = "solver.prototxt"
solverstate = "models-12/_iter_5000.solverstate"
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.get_solver(solver_file)
solver.restore(solverstate) #就用这一句
solver.solve()