简述:
相信跑过caffe框架的同学都知道
一般进行训练,我们都需要如下几个文件/文件夹:
1)models文件夹 就是存放训练得到的模型,也就是保存网络中的各种W和b的参数
2)train-SE.sh 其实就是一个脚本文件,里面写上类似这样的:
/home/lincanran/caffe/.build_release/tools/caffe train -solver=/home/lincanran/cls_exp2/SE-ResNeXt-101-2/SE-ResNeXt-101_solver.prototxt -weights=/home/lincanran/cls_exp2/SE-ResNeXt-101-2/SE-ResNeXt-101.caffemodel --gpu 3
这样我们就没必要每次在终端输入这么长的指令去执行训练操作了,只需要进入到.sh文件所在的目录,然后执行:
sudo sh train-SE.sh
即可
3)SE-ResNeXt-101_solver.prototxt 就是跑模型的参数配置文件,下面会详细介绍
4)SE-ResNeXt-101.prototxt 就是定义网络模型结构的文件,一般都是由很多layer组成的
5)SE-ResNeXt-101.caffemodel 就是预训练好的模型,一般在训