Caffe mnist数据集训练
2016-12-05 本文已影响70人
SnailTyan
文章作者:Tyan
博客:noahsnail.com | CSDN | 简书
本文是使用基于docker的caffe进行实验。docker启动过程请看docker相关的博客。训练过程如下:
# 进入caffe目录
$:/workspace# cd /opt/caffe/
# 下载mnist数据集
$:/opt/caffe# ./data/mnist/get_mnist.sh
# 将mnist数据集转换成lmdb
$:/opt/caffe# sh examples/mnist/create_mnist.sh
# 修改examples/mnist/lenet_solver.prototxt
将GPU改为CPU
# 运行examples/mnist/train_lenet.sh
$:/opt/caffe# sh examples/mnist/train_lenet.sh