name: "LeNet" layer { name: "mnist" type: "Data" top: "data" top: "label" include { phase: TRAIN } transform_param { scale: 0.00390625 } data_param { source: "mnist_train" 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: "mnist_test" batch_size: 100 backend: LMDB } } layer { name: "conv1" type: "Convolution" bottom: "mnist" top: "conv1" param { lr_mult: 1 } param { lr_mult: 2 } convolution_param { num_output: 32 kernel_size: 5 stride: 1 pad: 2 weight_filler { type: "msra" } bias_filler { type: "constant" value: 0.1 } } } layer { name: "relu1" type: "ReLU" bottom: "conv1" top: "relu1" } layer { name: "pool1" type: "Pooling" bottom: "relu1" top: "pool1" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "conv2" type: "Convolution" bottom: "pool1" top: "conv2" param { lr_mult: 1 } param { lr_mult: 2 } convolution_param { num_output: 64 kernel_size: 5 stride: 1 pad: 2 weight_filler { type: "msra" } bias_filler { type: "constant" value: 0.1 } } } layer { name: "relu2" type: "ReLU" bottom: "conv2" top: "relu2" } layer { name: "pool2" type: "Pooling" bottom: "relu2" top: "pool2" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "ip1" type: "InnerProduct" bottom: "pool2" top: "ip1" param { lr_mult: 1 } param { lr_mult: 2 } inner_product_param { num_output: 512 weight_filler { type: "msra" } bias_filler { type: "constant" value: 0 } } } layer { name: "relu3" type: "ReLU" bottom: "ip1" top: "relu3" } layer { name: "drop1" type: "Dropout" bottom: "relu3" top: "drop1" dropout_param { dropout_ratio: 0.5 } } layer { name: "ip2" type: "InnerProduct" bottom: "drop1" top: "ip2" param { lr_mult: 1 } param { lr_mult: 2 } inner_product_param { num_output: 10 weight_filler { type: "msra" } bias_filler { type: "constant" value: 0 } } } layer { name: "accuracy" type: "Accuracy" bottom: "ip2" bottom: "label" top: "accuracy" include { phase: TEST } } layer { name: "loss" type: "SoftmaxWithLoss" bottom: "ip2" bottom: "label" top: "loss" }