Fix error in mobilenetv1 for width multiplier < 1

FluxML · Apr 5, 2022 · edaed04 · edaed04
1 parent a94388d
commit edaed04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/convnets/mobilenet.jl b/src/convnets/mobilenet.jl
@@ -29,8 +29,8 @@ function mobilenetv1(width_mult, config;
                      fcsize = 1024)
   layers = []
   for (dw, outch, stride, repeats) in config
+    outch = Int(outch * width_mult)
     for _ in 1:repeats
-      outch = outch * width_mult
       layer = if dw
         depthwise_sep_conv_bn((3, 3), inchannels, outch, activation; stride = stride, pad = 1)
       else