Caffe mnist数据集训练

最新推荐文章于 2019-08-24 07:59:12 发布

原创最新推荐文章于 2019-08-24 07:59:12 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Caffe #mnist

深度学习同时被 2 个专栏收录

84 篇文章

订阅专栏

Caffe

24 篇文章

订阅专栏

本文介绍如何在Docker环境下使用Caffe框架训练MNIST数据集。具体步骤包括安装配置Docker环境、下载MNIST数据集并转换为LMDB格式、调整配置文件以适应CPU训练，并最终运行训练脚本。

文章作者：Tyan
博客：noahsnail.com | 优快云 | 简书

本文是使用基于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