File tree Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Expand file tree Collapse file tree 1 file changed +1
-4
lines changed Original file line number Diff line number Diff line change 4
4
mobilenetv1(width_mult, config;
5
5
activation = relu,
6
6
inchannels = 3,
7
- fcsize = 1024,
8
7
nclasses = 1000)
9
8
10
9
Create a MobileNetv1 model ([reference](https://arxiv.org/abs/1704.04861v1)).
@@ -28,7 +27,6 @@ Create a MobileNetv1 model ([reference](https://arxiv.org/abs/1704.04861v1)).
28
27
function mobilenetv1 (width_mult, config;
29
28
activation = relu,
30
29
inchannels = 3 ,
31
- fcsize = 1024 ,
32
30
nclasses = 1000 )
33
31
layers = []
34
32
for (dw, outch, stride, nrepeats) in config
@@ -47,8 +45,7 @@ function mobilenetv1(width_mult, config;
47
45
return Chain (Chain (layers),
48
46
Chain (GlobalMeanPool (),
49
47
MLUtils. flatten,
50
- Dense (inchannels, fcsize, activation),
51
- Dense (fcsize, nclasses)))
48
+ Dense (inchannels, nclasses)))
52
49
end
53
50
54
51
const mobilenetv1_configs = [
You can’t perform that action at this time.
0 commit comments