File tree Expand file tree Collapse file tree 1 file changed +125
-0
lines changed Expand file tree Collapse file tree 1 file changed +125
-0
lines changed Original file line number Diff line number Diff line change
1
+ name: "LeNet"
2
+ layers {
3
+ name: "data"
4
+ type: MEMORY_DATA
5
+ top: "data"
6
+ top: "label"
7
+ memory_data_param
8
+ {
9
+ batch_size: 10
10
+ channels: 1
11
+ height: 28
12
+ width: 28
13
+ }
14
+ transform_param {
15
+ scale: 0.00390625
16
+ }
17
+ }
18
+
19
+ layers {
20
+ name: "conv1"
21
+ type: CONVOLUTION
22
+ bottom: "data"
23
+ top: "conv1"
24
+ blobs_lr: 1
25
+ blobs_lr: 2
26
+ convolution_param {
27
+ num_output: 20
28
+ kernel_size: 5
29
+ stride: 1
30
+ weight_filler {
31
+ type: "xavier"
32
+ }
33
+ bias_filler {
34
+ type: "constant"
35
+ }
36
+ }
37
+ }
38
+ layers {
39
+ name: "pool1"
40
+ type: POOLING
41
+ bottom: "conv1"
42
+ top: "pool1"
43
+ pooling_param {
44
+ pool: MAX
45
+ kernel_size: 2
46
+ stride: 2
47
+ }
48
+ }
49
+ layers {
50
+ name: "conv2"
51
+ type: CONVOLUTION
52
+ bottom: "pool1"
53
+ top: "conv2"
54
+ blobs_lr: 1
55
+ blobs_lr: 2
56
+ convolution_param {
57
+ num_output: 50
58
+ kernel_size: 5
59
+ stride: 1
60
+ weight_filler {
61
+ type: "xavier"
62
+ }
63
+ bias_filler {
64
+ type: "constant"
65
+ }
66
+ }
67
+ }
68
+ layers {
69
+ name: "pool2"
70
+ type: POOLING
71
+ bottom: "conv2"
72
+ top: "pool2"
73
+ pooling_param {
74
+ pool: MAX
75
+ kernel_size: 2
76
+ stride: 2
77
+ }
78
+ }
79
+ layers {
80
+ name: "ip1"
81
+ type: INNER_PRODUCT
82
+ bottom: "pool2"
83
+ top: "ip1"
84
+ blobs_lr: 1
85
+ blobs_lr: 2
86
+ inner_product_param {
87
+ num_output: 500
88
+ weight_filler {
89
+ type: "xavier"
90
+ }
91
+ bias_filler {
92
+ type: "constant"
93
+ }
94
+ }
95
+ }
96
+ layers {
97
+ name: "relu1"
98
+ type: RELU
99
+ bottom: "ip1"
100
+ top: "ip1"
101
+ }
102
+ layers {
103
+ name: "ip2"
104
+ type: INNER_PRODUCT
105
+ bottom: "ip1"
106
+ top: "ip2"
107
+ blobs_lr: 1
108
+ blobs_lr: 2
109
+ inner_product_param {
110
+ num_output: 10
111
+ weight_filler {
112
+ type: "xavier"
113
+ }
114
+ bias_filler {
115
+ type: "constant"
116
+ }
117
+ }
118
+ }
119
+
120
+ layers {
121
+ name: "prob"
122
+ type: SOFTMAX
123
+ bottom: "ip2"
124
+ top: "prob"
125
+ }
You can’t perform that action at this time.
0 commit comments