Amend the weights only if quantize=True. (#4966)

b5670284 · Vasilis Vryniotis · GitHub · 8dcb5b81 · b5670284
Unverified Commit b5670284 authored Nov 21, 2021 by Vasilis Vryniotis Committed by GitHub Nov 21, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/models/quantization/mobilenetv3.py torchvision/models/quantization/mobilenetv3.py +1 -1

No files found.
--- a/torchvision/models/quantization/mobilenetv3.py
+++ b/torchvision/models/quantization/mobilenetv3.py
@@ -43,7 +43,7 @@ class QuantizableSqueezeExcitation(SqueezeExcitation):
    ):
        version = local_metadata.get("version", None)

-        if version is None or version < 2:
+        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.zero_point": torch.tensor([0], dtype=torch.int32),