fix mobilenetv3 quantization state dict loading (#4997)

58016b09 · Philip Meier · GitHub · 18cf5ab1 · 58016b09
Unverified Commit 58016b09 authored Nov 26, 2021 by Philip Meier Committed by GitHub Nov 26, 2021
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

torchvision/models/quantization/mobilenetv3.py torchvision/models/quantization/mobilenetv3.py +4 -0

No files found.
--- a/torchvision/models/quantization/mobilenetv3.py
+++ b/torchvision/models/quantization/mobilenetv3.py
@@ -46,7 +46,11 @@ class QuantizableSqueezeExcitation(SqueezeExcitation):
        if hasattr(self, "qconfig") and (version is None or version < 2):
            default_state_dict = {
                "scale_activation.activation_post_process.scale": torch.tensor([1.0]),
+                "scale_activation.activation_post_process.activation_post_process.scale": torch.tensor([1.0]),
                "scale_activation.activation_post_process.zero_point": torch.tensor([0], dtype=torch.int32),
+                "scale_activation.activation_post_process.activation_post_process.zero_point": torch.tensor(
+                    [0], dtype=torch.int32
+                ),
                "scale_activation.activation_post_process.fake_quant_enabled": torch.tensor([1]),
                "scale_activation.activation_post_process.observer_enabled": torch.tensor([1]),
            }