Caffe mnist数据集训练

文章作者: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

你可能感兴趣的:(Caffe mnist数据集训练)