Skip to content

Commit 5f50929

Browse files
committed
- Add lenet_test-memory-10.prototxt as example
1 parent c32bc42 commit 5f50929

File tree

1 file changed

+125
-0
lines changed

1 file changed

+125
-0
lines changed
Lines changed: 125 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,125 @@
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+
}

0 commit comments

Comments
 (0)