Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- Name: "MNIST Classification"
- layer {
- name: "data"
- type: "Data"
- top: "data"
- top: "label"
- transform_param {
- scale: 0.00392156862745
- }
- data_param {
- source: "mnist/mnist_train_lmdb"
- batch_size: 64
- backend: LMDB
- }
- }
- layer {
- name: "conv1"
- type: "Convolution"
- bottom: "data"
- top: "conv1"
- convolution_param {
- num_output: 20
- kernel_size: 5
- weight_filler {
- type: "xavier"
- }
- }
- }
- layer {
- name: "pool1"
- type: "Pooling"
- bottom: "conv1"
- top: "pool1"
- pooling_param {
- pool: MAX
- kernel_size: 2
- stride: 2
- }
- }
- layer {
- name: "conv2"
- type: "Convolution"
- bottom: "pool1"
- top: "conv2"
- convolution_param {
- num_output: 50
- kernel_size: 5
- weight_filler {
- type: "xavier"
- }
- }
- }
- layer {
- name: "pool2"
- type: "Pooling"
- bottom: "conv2"
- top: "pool2"
- pooling_param {
- pool: MAX
- kernel_size: 2
- stride: 2
- }
- }
- layer {
- name: "fc1"
- type: "InnerProduct"
- bottom: "pool2"
- top: "fc1"
- inner_product_param {
- num_output: 500
- weight_filler {
- type: "xavier"
- }
- }
- }
- layer {
- name: "relu1"
- type: "ReLU"
- bottom: "fc1"
- top: "fc1"
- }
- layer {
- name: "score"
- type: "InnerProduct"
- bottom: "fc1"
- top: "score"
- inner_product_param {
- num_output: 10
- weight_filler {
- type: "xavier"
- }
- }
- }
- layer {
- name: "loss"
- type: "SoftmaxWithLoss"
- bottom: "score"
- bottom: "label"
- top: "loss"
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement