fix

humu789 · Dec 21, 2022 · 297a2ab · 297a2ab
1 parent 9d66dd5
commit 297a2ab
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 12 deletions.
diff --git a/configs/quantization/ptq/ptq_openvino_mbv2_8xb32_in1k_calib32xb32.py b/configs/quantization/ptq/ptq_openvino_mbv2_8xb32_in1k_calib32xb32.py
@@ -21,8 +21,8 @@
     _delete_=True,
     type='mmrazor.MMArchitectureQuant',
     architecture=_base_.model,
-    # float_checkpoint='/tmp/humu/mobilenet_v2_batch256_imagenet' +
-    # '_20200708-3b2dc3af.pth',
+    float_checkpoint='/tmp/humu/mobilenet_v2_batch256_imagenet' +
+    '_20200708-3b2dc3af.pth',
     quantizer=dict(
         type='mmrazor.OpenVINOQuantizer',
         global_qconfig=global_qconfig,

diff --git a/configs/quantization/ptq/ptq_openvino_resnet50_8xb32_in1k_calib32xb32.py b/configs/quantization/ptq/ptq_openvino_resnet50_8xb32_in1k_calib32xb32.py
@@ -23,7 +23,7 @@
     _delete_=True,
     type='mmrazor.MMArchitectureQuant',
     architecture=_base_.model,
-    # float_checkpoint='/tmp/humu/resnet50_8xb32_in1k_20210831-ea4938fc.pth',
+    float_checkpoint='/tmp/humu/resnet50_8xb32_in1k_20210831-ea4938fc.pth',
     quantizer=dict(
         type='mmrazor.OpenVINOQuantizer',
         global_qconfig=global_qconfig,

diff --git a/mmrazor/models/quantizers/openvino_quantizer.py b/mmrazor/models/quantizers/openvino_quantizer.py
@@ -76,15 +76,6 @@ def prepare(self, model, graph_module):
             prepared = del_fakequant_after_target(
                 prepared, target_next, inplace=True)
         print(prepared)
-        cnt = 0
-
-        from ..fake_quants import BaseFakeQuantize
-        for name, module in prepared.named_modules():
-            if isinstance(module, BaseFakeQuantize) and 'weight' in name:
-                cnt += 1
-                # print()
-        print(cnt)
-        input()
 
         return prepared