-
Notifications
You must be signed in to change notification settings - Fork 40
/
alex_v2_1gpu.json
110 lines (110 loc) · 3.04 KB
/
alex_v2_1gpu.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
{
"name": "cuda-convnet2",
"layers": {
"data": {
"parents": [],
"type": "Input",
"tensor": [128, 224, 224, 3]
},
"conv1": {
"parents": ["data"],
"type": "Convolution",
"filter": [11, 11, 3, 64],
"padding": "VALID",
"strides": [1, 4, 4, 1],
"activation_fn": "relu"
},
"pool1": {
"parents": ["conv1"],
"type": "Pooling",
"ksize": [1, 3, 3, 1],
"strides": [1, 2, 2, 1],
"padding": "VALID"
},
"conv2": {
"parents": ["pool1"],
"type": "Convolution",
"filter": [5, 5, 64, 192],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"pool2": {
"parents": ["conv2"],
"type": "Pooling",
"ksize": [1, 3, 3, 1],
"strides": [1, 2, 2, 1],
"padding": "VALID"
},
"conv3": {
"parents": ["pool2"],
"type": "Convolution",
"filter": [3, 3, 192, 384],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"conv4": {
"parents": ["conv3"],
"type": "Convolution",
"filter": [3, 3, 384, 256],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"conv5": {
"parents": ["conv4"],
"type": "Convolution",
"filter": [3, 3, 256, 256],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"pool5": {
"parents": ["conv5"],
"type": "Pooling",
"ksize": [1, 3, 3, 1],
"strides": [1, 2, 2, 1],
"padding": "VALID"
},
"fc6": {
"parents": ["pool5"],
"type": "Convolution",
"filter": [5, 5, 256, 4096],
"padding": "VALID",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"dropout6": {
"parents": ["fc6"],
"type": "Dropout",
"dropout_keep_prob": 0.5
},
"fc7": {
"parents": ["dropout6"],
"type": "Convolution",
"filter": [1, 1, 4096, 4096],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": "relu"
},
"dropout7": {
"parents": ["fc7"],
"type": "Dropout",
"dropout_keep_prob": 0.5
},
"fc8": {
"parents": ["dropout7"],
"type": "Convolution",
"filter": [1, 1, 4096, 1000],
"padding": "SAME",
"strides": [1, 1, 1, 1],
"activation_fn": null
},
"softmax": {
"parents": ["fc8"],
"type": "Softmax",
"num_classes": 1000
}
}
}