Skip to content

Commit 134e316

Browse files
committed
removed extra mobilenetv1 fc layer, fcsize param
1 parent 48d1401 commit 134e316

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

src/convnets/mobilenet.jl

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
mobilenetv1(width_mult, config;
55
activation = relu,
66
inchannels = 3,
7-
fcsize = 1024,
87
nclasses = 1000)
98
109
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)).
2827
function mobilenetv1(width_mult, config;
2928
activation = relu,
3029
inchannels = 3,
31-
fcsize = 1024,
3230
nclasses = 1000)
3331
layers = []
3432
for (dw, outch, stride, nrepeats) in config
@@ -47,8 +45,7 @@ function mobilenetv1(width_mult, config;
4745
return Chain(Chain(layers),
4846
Chain(GlobalMeanPool(),
4947
MLUtils.flatten,
50-
Dense(inchannels, fcsize, activation),
51-
Dense(fcsize, nclasses)))
48+
Dense(inchannels, nclasses)))
5249
end
5350

5451
const mobilenetv1_configs = [

0 commit comments

Comments
 (0)