Yodazon commited on
Commit
3c2be94
·
verified ·
1 Parent(s): ea94aba

Upload config.js

Browse files
Files changed (1) hide show
  1. config.js +126 -0
config.js ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "pyTorchModel",
3
+ "num_classes": 4,
4
+ "layers": [
5
+ {
6
+ "type": "Conv2d",
7
+ "in_channels": 3,
8
+ "out_channels": 96,
9
+ "kernel_size": 11,
10
+ "stride": 4,
11
+ "padding": 0
12
+ },
13
+ {
14
+ "type": "BatchNorm2d",
15
+ "num_features": 96
16
+ },
17
+ {
18
+ "type": "ReLU"
19
+ },
20
+ {
21
+ "type": "MaxPool2d",
22
+ "kernel_size": 3,
23
+ "stride": 2
24
+ },
25
+ {
26
+ "type": "Conv2d",
27
+ "in_channels": 96,
28
+ "out_channels": 256,
29
+ "kernel_size": 5,
30
+ "stride": 1,
31
+ "padding": 2
32
+ },
33
+ {
34
+ "type": "BatchNorm2d",
35
+ "num_features": 256
36
+ },
37
+ {
38
+ "type": "ReLU"
39
+ },
40
+ {
41
+ "type": "MaxPool2d",
42
+ "kernel_size": 3,
43
+ "stride": 2
44
+ },
45
+ {
46
+ "type": "Conv2d",
47
+ "in_channels": 256,
48
+ "out_channels": 384,
49
+ "kernel_size": 3,
50
+ "stride": 1,
51
+ "padding": 1
52
+ },
53
+ {
54
+ "type": "BatchNorm2d",
55
+ "num_features": 384
56
+ },
57
+ {
58
+ "type": "ReLU"
59
+ },
60
+ {
61
+ "type": "Conv2d",
62
+ "in_channels": 384,
63
+ "out_channels": 384,
64
+ "kernel_size": 3,
65
+ "stride": 1,
66
+ "padding": 1
67
+ },
68
+ {
69
+ "type": "BatchNorm2d",
70
+ "num_features": 384
71
+ },
72
+ {
73
+ "type": "ReLU"
74
+ },
75
+ {
76
+ "type": "Conv2d",
77
+ "in_channels": 384,
78
+ "out_channels": 256,
79
+ "kernel_size": 3,
80
+ "stride": 1,
81
+ "padding": 1
82
+ },
83
+ {
84
+ "type": "BatchNorm2d",
85
+ "num_features": 256
86
+ },
87
+ {
88
+ "type": "ReLU"
89
+ },
90
+ {
91
+ "type": "MaxPool2d",
92
+ "kernel_size": 3,
93
+ "stride": 2
94
+ },
95
+ {
96
+ "type": "Dropout",
97
+ "p": 0.5
98
+ },
99
+ {
100
+ "type": "Linear",
101
+ "in_features": 9216,
102
+ "out_features": 4096
103
+ },
104
+ {
105
+ "type": "ReLU"
106
+ },
107
+ {
108
+ "type": "Dropout",
109
+ "p": 0.5
110
+ },
111
+ {
112
+ "type": "Linear",
113
+ "in_features": 4096,
114
+ "out_features": 4096
115
+ },
116
+ {
117
+ "type": "ReLU"
118
+ },
119
+ {
120
+ "type": "Linear",
121
+ "in_features": 4096,
122
+ "out_features": 4
123
+ }
124
+ ]
125
+ }
126
+