Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- name: "simple_net"
- layer {
- name: "data"
- type: "Input"
- top: "data"
- include {
- phase: TRAIN
- }
- input_param {
- shape {
- dim: 100
- dim: 1
- dim: 28
- dim: 28
- }
- }
- }
- layer {
- name: "data"
- type: "Input"
- top: "data"
- include {
- phase: TEST
- }
- input_param {
- shape {
- dim: 100
- dim: 1
- dim: 28
- dim: 28
- }
- }
- }
- layer {
- name: "label"
- type: "Input"
- top: "label"
- include {
- phase: TRAIN
- }
- input_param {
- shape {
- dim: 100
- }
- }
- }
- layer {
- name: "label"
- type: "Input"
- top: "label"
- include {
- phase: TEST
- }
- input_param {
- shape {
- dim: 100
- }
- }
- }
- layer {
- name: "conv1"
- type: "Convolution"
- bottom: "data"
- top: "conv1"
- param {
- lr_mult: 1.0
- }
- param {
- lr_mult: 2.0
- }
- convolution_param {
- num_output: 32
- kernel_size: 5
- weight_filler {
- type: "xavier"
- }
- bias_filler {
- type: "constant"
- }
- }
- }
- layer {
- name: "relu1"
- type: "ReLU"
- bottom: "conv1"
- top: "conv1"
- }
- layer {
- name: "conv2"
- type: "Convolution"
- bottom: "conv1"
- top: "conv2"
- param {
- lr_mult: 1.0
- }
- param {
- lr_mult: 2.0
- }
- convolution_param {
- num_output: 32
- kernel_size: 5
- weight_filler {
- type: "xavier"
- }
- bias_filler {
- type: "constant"
- }
- }
- }
- layer {
- name: "relu2"
- type: "ReLU"
- bottom: "conv2"
- top: "conv2"
- }
- layer {
- name: "fc1"
- type: "InnerProduct"
- bottom: "conv2"
- top: "fc1"
- param {
- lr_mult: 1.0
- }
- param {
- lr_mult: 2.0
- }
- inner_product_param {
- num_output: 500
- weight_filler {
- type: "xavier"
- }
- bias_filler {
- type: "constant"
- }
- }
- }
- layer {
- name: "relu3"
- type: "ReLU"
- bottom: "fc1"
- top: "fc1"
- }
- layer {
- name: "out"
- type: "InnerProduct"
- bottom: "fc1"
- top: "out"
- param {
- lr_mult: 1.0
- }
- param {
- lr_mult: 2.0
- }
- inner_product_param {
- num_output: 10
- weight_filler {
- type: "xavier"
- }
- bias_filler {
- type: "constant"
- }
- }
- }
- layer {
- name: "loss"
- type: "SoftmaxWithLoss"
- bottom: "out"
- bottom: "label"
- top: "loss"
- }
- layer {
- name: "accuracy"
- type: "Accuracy"
- bottom: "out"
- bottom: "label"
- top: "accuracy"
- include {
- phase: TEST
- }
- }
Add Comment
Please, Sign In to add comment