Skip to content

Commit 1851e55

Browse files
committed
Fixing typo.
1 parent b491fa2 commit 1851e55

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/models/quantization/mobilenetv3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ class QuantizableSqueezeExcitation(SElayer):
2121
_version = 2
2222

2323
def __init__(self, *args: Any, **kwargs: Any) -> None:
24-
kwargs["scale_activation"] = nn.Hardswish
24+
kwargs["scale_activation"] = nn.Hardsigmoid
2525
super().__init__(*args, **kwargs)
2626
self.skip_mul = nn.quantized.FloatFunctional()
2727

0 commit comments

Comments
 (0)