Deep Learning---基于windows版caffe训练mnist数据集

来源:互联网 发布:阿里云域名接入 编辑:程序博客网 时间:2024/05/24 07:38

1 训练前准备

  • 下载数据集:mnist-leveldb.7z(http://pan.baidu.com/s/1qY5857e)
  • 解压到caffe-master\examples\mnist目录下
  • 配置mnist_autoencoder_solver.prototxt和lenet_train_test.prototxt以及lenet_adadelta_solver.prototxt

mnist_autoencoder_solver.prototxt

lenet_train_test.prototxt

lenet_adadelta_solver.prototxt

2 训练

在cmd命令窗口中进入到caffe.exe的目录,输入命令:

caffe train -solver=E:\caffe-master\examples\mnist\lenet_adadelta_solver.prototxt

train

最后训练完成生成模型:

models

lenet_adadelta_iter_10000.caffemodel是迭代10000次后生成的caffe model
lenet_adadelta_iter_5000.caffemodel是迭代5000次后生成的caffe model

0 0