import caffe solver_file = "solver.prototxt" solverstate = "xx.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 = "xx.solverstate" caffe.set_device(0) caffe.set_mode_gpu() solver = caffe.get_solver(solver_file) solver.restore(solverstate) solver.solve()
转载于:https://www.cnblogs.com/houjun/p/9819935.html