Caffe程序员@IT·互联网

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
上一篇下一篇

猜你喜欢

热点阅读