diff --git a/demo/dygraph/quant/mobilenet_v3.py b/demo/dygraph/quant/mobilenet_v3.py index e56c8990995a35dd6587013955322bf3873d1aa2..f9e6996283acfdebea623d395ce844ad897e71cf 100644 --- a/demo/dygraph/quant/mobilenet_v3.py +++ b/demo/dygraph/quant/mobilenet_v3.py @@ -21,7 +21,6 @@ import paddle from paddle import ParamAttr import paddle.nn as nn import paddle.nn.functional as F -from paddle.nn.functional.activation import hard_sigmoid, hard_swish from paddle.nn import Conv2D, BatchNorm, Linear, Dropout from paddle.nn import AdaptiveAvgPool2D, MaxPool2D, AvgPool2D from paddle.regularizer import L2Decay @@ -165,7 +164,7 @@ class MobileNetV3(nn.Layer): x = self.pool(x) x = self.last_conv(x) - x = hard_swish(x) + x = paddle.nn.functional.activation.hardswish(x) x = paddle.reshape(x, shape=[x.shape[0], x.shape[1]]) x = self.out(x) @@ -303,7 +302,8 @@ class SEModule(nn.Layer): outputs = self.conv1(outputs) outputs = F.relu(outputs) outputs = self.conv2(outputs) - outputs = hard_sigmoid(outputs) + outputs = paddle.nn.functional.activation.hardsigmoid( + outputs, slope=0.2) return paddle.multiply(x=inputs, y=outputs)