a7d578efddfcbee46302b0892a51952ccb7d4b03
[jacinto-ai/caffe-jacinto.git] / src / caffe / test / data / lenet.prototxt
1 name: "LeNet"
2 bottom: "data"
3 bottom: "label"
4 layers {
5   layer {
6     name: "conv1"
7     type: "conv"
8     num_output: 20
9     kernelsize: 5
10     stride: 1
11     weight_filler {
12       type: "xavier"
13     }
14     bias_filler {
15       type: "constant"
16     }
17   }
18   bottom: "data"
19   top: "conv1"
20 }
21 layers {
22   layer {
23     name: "pool1"
24     type: "pool"
25     kernelsize: 2
26     stride: 2
27     pool: MAX
28   }
29   bottom: "conv1"
30   top: "pool1"
31 }
32 layers {
33   layer {
34     name: "conv2"
35     type: "conv"
36     num_output: 50
37     kernelsize: 5
38     stride: 1
39     weight_filler {
40       type: "xavier"
41     }
42     bias_filler {
43       type: "constant"
44     }
45   }
46   bottom: "pool1"
47   top: "conv2"
48 }
49 layers {
50   layer {
51     name: "pool2"
52     type: "pool"
53     kernelsize: 2
54     stride: 2
55     pool: MAX
56   }
57   bottom: "conv2"
58   top: "pool2"
59 }
60 layers {
61   layer {
62     name: "ip1"
63     type: "innerproduct"
64     num_output: 500
65     weight_filler {
66       type: "xavier"
67     }
68     bias_filler {
69       type: "constant"
70     }
71   }
72   bottom: "pool2"
73   top: "ip1"
74 }
75 layers {
76   layer {
77     name: "relu1"
78     type: "relu"
79   }
80   bottom: "ip1"
81   top: "relu1"
82 }
83 layers {
84   layer {
85     name: "ip2"
86     type: "innerproduct"
87     num_output: 10
88     weight_filler {
89       type: "xavier"
90     }
91     bias_filler {
92       type: "constant"
93     }
94   }
95   bottom: "relu1"
96   top: "ip2"
97 }
98 layers {
99   layer {
100     name: "prob"
101     type: "softmax"
102   }
103   bottom: "ip2"
104   top: "prob"
106 layers {
107   layer {
108     name: "loss"
109     type: "multinomial_logistic_loss"
110   }
111   bottom: "prob"
112   bottom: "label"