caffe + windows +cpu +vs2013安装与minist测试(中)

来源:互联网 发布:兴达驾校网络预约系统 编辑:程序博客网 时间:2024/05/19 04:56

修改网络模型表述文件

F:\caffe0\caffe-windows\examples\mnist\lenet_train_test.prototxt

目前主要修改训练集和测试集的路径 注意路径的斜杠表示

name: "LeNet"layer {  name: "mnist"  type: "Data"  top: "data"  top: "label"  include {    phase: TRAIN  }  transform_param {    scale: 0.00390625  }  data_param {    source: "F:/caffe0/caffe-windows/examples/mnist/LMDB/train_lmdb"    batch_size: 64    backend: LMDB  }}layer {  name: "mnist"  type: "Data"  top: "data"  top: "label"  include {    phase: TEST  }  transform_param {    scale: 0.00390625  }  data_param {    source: "F:/caffe0/caffe-windows/examples/mnist/LMDB/test_lmdb"    batch_size: 100    backend: LMDB  }}

修改超参数

主要修改网络模型描述文件的路径
训练输出参数
此处选择CPU

# The train/test net protocol buffer definitionnet: "F:/caffe0/caffe-windows/examples/mnist/lenet_train_test.prototxt"# test_iter specifies how many forward passes the test should carry out.# In the case of MNIST, we have test batch size 100 and 100 test iterations,# covering the full 10,000 testing images.test_iter: 100# Carry out testing every 500 training iterations.test_interval: 500# The base learning rate, momentum and the weight decay of the network.base_lr: 0.01momentum: 0.9weight_decay: 0.0005# The learning rate policylr_policy: "inv"gamma: 0.0001power: 0.75# Display every 100 iterationsdisplay: 100# The maximum number of iterationsmax_iter: 10000# snapshot intermediate resultssnapshot: 5000snapshot_prefix: "F:/caffe0/caffe-windows/examples/mnist/models/"  注意/斜杠# solver mode: CPU or GPUsolver_mode: CPU

开始训练模型

新建批处理文件 train.bat
F:\caffe0\caffe-windows\tools\caffe.cpp

第一项为可执行文件路径
第二项为 train 或者test
第三项为超参数文件的路径

%train 训练数据%F:\caffe0\caffe-windows\Build\x64\Debug\caffe.exe train -solver=F:/caffe0/caffe-windows/examples/mnist/lenet_solver.prototxt

深度学习第一天,写的不深入以后再补充,才疏学浅,有错误或者不足,欢迎指正!

阅读全文
0 0
原创粉丝点击