Fix the api error of mobilenetv3 (PaddlePaddle#1256)

ETTR123 · Aug 17, 2021 · 37d7745 · 37d7745
1 parent c8acc1d
commit 37d7745
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddleseg/models/backbones/mobilenetv3.py b/paddleseg/models/backbones/mobilenetv3.py
@@ -308,8 +308,8 @@ def forward(self, inputs):
         outputs = self.conv1(outputs)
         outputs = F.relu(outputs)
         outputs = self.conv2(outputs)
-        outputs = F.hard_sigmoid(outputs)
-        return paddle.multiply(x=inputs, y=outputs, axis=0)
+        outputs = F.hardsigmoid(outputs)
+        return paddle.multiply(x=inputs, y=outputs)
 
 
 def MobileNetV3_small_x0_35(**kwargs):