From 4bb065df015925d49fd1acaa58e1f1ac5ffffc8a Mon Sep 17 00:00:00 2001 From: Abhirath Anand <74202102+theabhirath@users.noreply.github.com> Date: Sat, 25 Jun 2022 14:34:48 +0530 Subject: [PATCH] Formatting etc Co-authored-by: Kyle Daruwalla --- src/convnets/mobilenet.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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