From e63e886140b391ced481658f99f8aa06aa14f6b2 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Sat, 20 Nov 2021 11:19:41 +0000 Subject: [PATCH] Amend the weights only if quantize=True. --- torchvision/models/quantization/mobilenetv3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/quantization/mobilenetv3.py b/torchvision/models/quantization/mobilenetv3.py index 948b72ead7d..6eaa9114d74 100644 --- a/torchvision/models/quantization/mobilenetv3.py +++ b/torchvision/models/quantization/mobilenetv3.py @@ -43,7 +43,7 @@ def _load_from_state_dict( ): 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),