diff --git a/src/convnets/mobilenet.jl b/src/convnets/mobilenet.jl index d1fb2fd0a..93eba1c06 100644 --- a/src/convnets/mobilenet.jl +++ b/src/convnets/mobilenet.jl @@ -4,6 +4,7 @@ mobilenetv1(width_mult, config; activation = relu, inchannels = 3, + fcsize = 1024, nclasses = 1000) Create a MobileNetv1 model ([reference](https://arxiv.org/abs/1704.04861v1)). @@ -27,7 +28,7 @@ Create a MobileNetv1 model ([reference](https://arxiv.org/abs/1704.04861v1)). function mobilenetv1(width_mult, config; activation = relu, inchannels = 3, - fcsize = 1024, + fcsize = 1024, nclasses = 1000) layers = [] for (dw, outch, stride, nrepeats) in config