Merge pull request #265 from darsnack/mobilenet-constructor-fix

Fix bug where MobileNetv2/3 return a MobileNetv1
FluxML · Dec 22, 2023 · 3bd48ad · 3bd48ad
2 parents 268da20 + 3665a11
commit 3bd48ad
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "Metalhead"
 uuid = "dbeba491-748d-5e0e-a39e-b530a07fa0cc"
-version = "0.9.0"
+version = "0.9.1"
 
 [deps]
 Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33"

diff --git a/src/convnets/mobilenets/mobilenetv2.jl b/src/convnets/mobilenets/mobilenetv2.jl
@@ -71,7 +71,7 @@ end
 function MobileNetv2(width_mult::Real = 1; pretrain::Bool = false,
                      inchannels::Integer = 3, nclasses::Integer = 1000)
     layers = mobilenetv2(width_mult; inchannels, nclasses)
-    model = MobileNetv1(layers)
+    model = MobileNetv2(layers)
     if pretrain
         loadpretrain!(model, "mobilenet_v2")
     end

diff --git a/src/convnets/mobilenets/mobilenetv3.jl b/src/convnets/mobilenets/mobilenetv3.jl
@@ -91,7 +91,7 @@ end
 function MobileNetv3(config::Symbol; width_mult::Real = 1, pretrain::Bool = false,
                      inchannels::Integer = 3, nclasses::Integer = 1000)
     layers = mobilenetv3(config; width_mult, inchannels, nclasses)
-    model = MobileNetv1(layers)
+    model = MobileNetv3(layers)
     if pretrain
         loadpretrain!(model, "mobilenet_v3")
     end