use sigmoid mask

Tencent · Jul 28, 2022 · c405dfd · c405dfd
1 parent a2bb826
commit c405dfd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tools/pnnx/tests/ncnn/test_torchvision_DeformConv2d.py b/tools/pnnx/tests/ncnn/test_torchvision_DeformConv2d.py
@@ -31,7 +31,7 @@ def forward(self, x):
         offset = self.conv_0(x)
         x1 = self.conv_1(x, offset)
 
-        mask = self.conv_2(x)
+        mask = F.sigmoid(self.conv_2(x))
         x2 = self.conv_3(x, offset, mask)
         return x1, x2
 

diff --git a/tools/pnnx/tests/test_torchvision_DeformConv2d.py b/tools/pnnx/tests/test_torchvision_DeformConv2d.py
@@ -31,7 +31,7 @@ def forward(self, x):
         offset = self.conv_0(x)
         x1 = self.conv_1(x, offset)
 
-        mask = self.conv_2(x)
+        mask = F.sigmoid(self.conv_2(x))
         x2 = self.conv_3(x, offset, mask)
         return x1, x2