在ubuntu16.04中如何使用caffe训练自带的mnist数据集

首先需要注意两条:

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值