caffe mnist训练

版权声明:ˋ( ° ▽、° )  ̄へ ̄ https://blog.csdn.net/acm_fighting/article/details/60965746

1.数据集的准备

在caffe根目录下运行

./data/mnist/get_mnist.sh   #获得mnist数据训练的四个数据包
./examples/siamese/create_mnist_siamese.sh  #将mnist date转化为可用的lmdb格式的文件,并将新生成的2个文件mnist-train-lmdb 和 mnist-test-lmdb放于create_mnist.sh同目录下。

2.训练数据

在caffe根目录下运行

./examples/mnist/train_lenet.sh  

过程中产生的错误

I0309 18:44:30.833485 3083 caffe.cpp:218] Using GPUs 0
F0309 18:44:30.833644 3083 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode.

caffe_option(CPU_ONLY “Build Caffe without CUDA support” OFF) # TODO: rename to USE_CUDA


解决方案

之前选的是CPU-ONLY
编辑examples/mnist文件夹下的lenet_solver.prototxt文件,将solver_mode模式从GPU改为CPU.

展开阅读全文

没有更多推荐了,返回首页