diff --git a/python/mxnet/gluon/contrib/cnn/conv_layers.py b/python/mxnet/gluon/contrib/cnn/conv_layers.py index 3a30f59de44f..098463eca968 100644 --- a/python/mxnet/gluon/contrib/cnn/conv_layers.py +++ b/python/mxnet/gluon/contrib/cnn/conv_layers.py @@ -379,7 +379,7 @@ def hybrid_forward(self, F, x, offset_weight, deformable_conv_weight, offset_bia offset_t = F.slice_axis(offset, axis=1, begin=0, end=18) mask = F.slice_axis(offset, axis=1, begin=18, end=None) - mask = F.sigmoid(mask) + mask = F.sigmoid(mask) * 2 if deformable_conv_bias is None: act = F.contrib.ModulatedDeformableConvolution(data=x, offset=offset_t, mask=mask,