Make sure groups keyword is an integer

FluxML · Apr 5, 2022 · daba9a8 · daba9a8
1 parent 8f6cc77
commit daba9a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layers/conv.jl b/src/layers/conv.jl
@@ -85,7 +85,7 @@ depthwise_sep_conv_bn(kernelsize, inplanes, outplanes, activation = relu;
   vcat(conv_bn(kernelsize, inplanes, inplanes, activation;
                rev = rev, initβ = initβ, initγ = initγ,
                ϵ = ϵ, momentum = momentum,
-               stride = stride, groups = inplanes, kwargs...),
+               stride = stride, groups = Int(inplanes), kwargs...),
        conv_bn((1, 1), inplanes, outplanes, activation;
                rev = rev, initβ = initβ, initγ = initγ,
                ϵ = ϵ, momentum = momentum))