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.
转载请注明原文地址: https://ju.6miu.com/read-15252.html
