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()