fix

openvinotoolkit · Oct 12, 2023 · 8d1beb2 · 8d1beb2
1 parent 07cec83
commit 8d1beb2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/...lassification/configs/mixed_precision/squeezenet1_1_imagenet_mixed_int_hawq_old_eval.json b/...lassification/configs/mixed_precision/squeezenet1_1_imagenet_mixed_int_hawq_old_eval.json
@@ -9,5 +9,6 @@
     "target_device": "TRIAL",
     "compression": {
         "algorithm": "quantization"
-    }
+    },
+    "no_strip_on_export": true
 }
diff --git a/examples/torch/classification/staged_quantization_worker.py b/examples/torch/classification/staged_quantization_worker.py
@@ -212,7 +212,7 @@ def autoq_eval_fn(model, eval_loader):
 
     best_acc1 = 0
     # optionally resume from a checkpoint
-    if resuming_checkpoint is not None and config.to_onnx is None:
+    if resuming_checkpoint is not None and config.to_ir is None:
         best_acc1 = resuming_checkpoint["best_acc1"]
         if "train" in config.mode:
             kd_loss_calculator.original_model.load_state_dict(resuming_checkpoint["original_model_state_dict"])