From 3abfbd2fa04165f3012752298a561e2d9dd95f04 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Fri, 26 Nov 2021 12:31:41 +0100 Subject: [PATCH] fix mobilenetv3 quantization state dict loading --- torchvision/models/quantization/mobilenetv3.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/torchvision/models/quantization/mobilenetv3.py b/torchvision/models/quantization/mobilenetv3.py index 6eaa9114d74..c831a443d78 100644 --- a/torchvision/models/quantization/mobilenetv3.py +++ b/torchvision/models/quantization/mobilenetv3.py @@ -46,7 +46,11 @@ def _load_from_state_dict( 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]), }