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.