首先需要注意两条:
1, 脚本的运行基于$Caffe_Root文件夹下的路径执行,
2,训练的时候,如果安装的时候选择了CPU_ONLY的话,在lenet_solver..prototxt文件中,把“mode:GPU”改成“mode:CPU”
此时便可开始训练mnist数据了
第一步:. 获取数据
./data/mnist/get_mnist.sh
第二步: 将数据转化为lmdb格式
./examples/mnist/create_mnist.sh
第三步:. 训练
./examples/mnist/train_lenet.sh
此时若出现
Check failed: mdb_status==0(13 vs. 0) Permission denied
这类错误,说明权限不够,可以把第三步改成:
sudo ./examples/mnist/train_lenet.sh