If you don't want to visualize and just want to get a final training model, then the code is very simple, as follows:
import caffe
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.SGDSolver('/home/xxx/data/solver.prototxt') solver.solve()
If you don't want to visualize and just want to get a final training model, then the code is very simple, as follows:
import caffe
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.SGDSolver('/home/xxx/data/solver.prototxt') solver.solve()