We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b491fa2 commit 1851e55Copy full SHA for 1851e55
torchvision/models/quantization/mobilenetv3.py
@@ -21,7 +21,7 @@ class QuantizableSqueezeExcitation(SElayer):
21
_version = 2
22
23
def __init__(self, *args: Any, **kwargs: Any) -> None:
24
- kwargs["scale_activation"] = nn.Hardswish
+ kwargs["scale_activation"] = nn.Hardsigmoid
25
super().__init__(*args, **kwargs)
26
self.skip_mul = nn.quantized.FloatFunctional()
27
0 commit comments