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 c...
原创
2018-08-02 14:49:05 ·
410 阅读 ·
0 评论