在ubuntu16.04中如何使用caffe训练自带的mnist数据集
首先需要注意两条:1, 脚本的运行基于$Caffe_Root文件夹下的路径执行,2,训练的时候,如果安装的时候选择了CPU_ONLY的话,在lenet_solver..prototxt文件中,把“mode:GPU”改成“mode:CPU”此时便可开始训练mnist数据了第一步:. 获取数据./data/mnist/get_mnist.sh第二步: 将数据转化为lmdb格式
原创
2017-11-03 09:29:54 ·
559 阅读 ·
0 评论