Fixes

Author: theabhirath

src/convnets/efficientnets/efficientnetv2.jl

@@ -54,7 +54,8 @@ end
 function EfficientNetv2(config::Symbol; pretrain::Bool = false,
                         inchannels::Integer = 3, nclasses::Integer = 1000)
     _checkconfig(config, sort(collect(keys(EFFNETV2_CONFIGS))))
-    layers = efficientnet(EFFNETV2_CONFIGS[config]; inchannels, nclasses)
+    layers = efficientnet(EFFNETV2_CONFIGS[config]; headplanes = 1280, inchannels,
+                          nclasses)
     if pretrain
         loadpretrain!(layers, string("efficientnetv2"))
     end

src/layers/conv.jl

@@ -181,7 +181,8 @@ function fused_mbconv(kernel_size::Dims{2}, inplanes::Integer,
     if explanes != inplanes
         # fused expand
         append!(layers,
-                conv_norm(kernel_size, inplanes, explanes, activation; norm_layer, stride))
+                conv_norm(kernel_size, inplanes, explanes, activation; norm_layer, stride,
+                          pad = SamePad()))
         # project
         append!(layers, conv_norm((1, 1), explanes, outplanes, identity; norm_layer))
     else