感谢博主lynnandwei,写的博客。还是由于受其启发,所以才想到了怎么微调模型,哎,菜比伤不起。
还是在cifar10模型下面。想微调已经生成的模型。在caffe-master文件夹下面输入命令
./build/tools/caffe.bin train -model=examples/cifar10/cifar10_quick_train_test.prototxt
-solver=examples/cifar10/cifar10_quick_solver.prototxt
-weight=examples/cifar10/cifar10_quick_iter_5000.caffemodel -gpu=0
这里模型就是这网络的结构。配置就是配置文件,权重是初始化权重。由于配置文件里面直接是4000保存出一个模型。所以这里挺搞笑的。4000的那个模型发生了改变。5000的模型还是原样。
时间: 2024-10-11 15:55:55