使用Caffe框架训练,一般需要网络模型prototxt文件,还需要超参数文件solver.prototxt文件,大部分我们都使用fine-tuning的方法,即加-weights选项,后面跟上一个caffemodel.
./cafferead/build/tools/caffe train -model xxxx.prototxt -solver=xxxx_solver.prototxt -weights xxxx.caffemodel -gpu all/0
使用Caffe框架训练,一般需要网络模型prototxt文件,还需要超参数文件solver.prototxt文件,大部分我们都使用fine-tuning的方法,即加-weights选项,后面跟上一个caffemodel.
./cafferead/build/tools/caffe train -model xxxx.prototxt -solver=xxxx_solver.prototxt -weights xxxx.caffemodel -gpu all/0