-
Notifications
You must be signed in to change notification settings - Fork 40
/
nin.json
144 lines (144 loc) · 4.13 KB
/
nin.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
{
"name": "NIN - https://gist.github.com/mavenlin/d802a5849de39225bcc6#file-train_val-prototxt",
"layers": {
"data": {
"parents": [],
"type": "Input",
"tensor": [128, 224, 224, 3]
},
"conv1": {
"parents": ["data"],
"type": "Convolution",
"filter": [11, 11, 3, 96],
"padding": "VALID",
"strides": [1, 4, 4, 1],
"activation_fn": "relu"
},
"cccp1": {
"parents": ["conv1"],
"type": "Convolution",
"filter": [1, 1, 96, 96],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"cccp2": {
"parents": ["cccp1"],
"type": "Convolution",
"filter": [1, 1, 96, 96],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"pool0": {
"parents": ["cccp2"],
"type": "Pooling",
"ksize": [1, 3, 3, 1],
"strides": [1, 2, 2, 1],
"padding": "VALID"
},
"conv2": {
"parents": ["pool0"],
"type": "Convolution",
"filter": [5, 5, 96, 256],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"cccp3": {
"parents": ["conv2"],
"type": "Convolution",
"filter": [1, 1, 256, 256],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"cccp4": {
"parents": ["cccp3"],
"type": "Convolution",
"filter": [1, 1, 256, 256],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"pool2": {
"parents": ["cccp4"],
"type": "Pooling",
"ksize": [1, 3, 3, 1],
"strides": [1, 2, 2, 1],
"padding": "VALID"
},
"conv3": {
"parents": ["pool2"],
"type": "Convolution",
"filter": [3, 3, 256, 384],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"cccp5": {
"parents": ["conv3"],
"type": "Convolution",
"filter": [1, 1, 384, 384],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"cccp6": {
"parents": ["cccp5"],
"type": "Convolution",
"filter": [1, 1, 384, 384],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"pool3": {
"parents": ["cccp6"],
"type": "Pooling",
"ksize": [1, 3, 3, 1],
"strides": [1, 2, 2, 1],
"padding": "VALID"
},
"drop": {
"parents": ["pool3"],
"type": "Dropout",
"dropout_keep_prob": 0.5
},
"conv4": {
"parents": ["drop"],
"type": "Convolution",
"filter": [3, 3, 384, 1024],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"cccp7": {
"parents": ["conv4"],
"type": "Convolution",
"filter": [1, 1, 1024, 1024],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"cccp8": {
"parents": ["cccp7"],
"type": "Convolution",
"filter": [1, 1, 1024, 1000],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"pool4": {
"parents": ["cccp8"],
"type": "AvgPool",
"ksize": [1, 5, 5, 1],
"strides": [1, 1, 1, 1],
"padding": "VALID"
},
"softmax": {
"parents": ["pool4"],
"type": "Softmax",
"num_classes": 1000
}
}
}