From bdc316fd89de91936951cb15472a01d620c13467 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Wed, 28 Aug 2024 10:19:02 +0100 Subject: [PATCH] fix ptq script --- .../imagenet_classification/ptq/ptq_evaluate.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py index 8fd2f655d..3a9bb29fa 100644 --- a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py +++ b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py @@ -18,8 +18,6 @@ from brevitas.export import export_onnx_qcdq from brevitas.export import export_torch_qcdq -from brevitas.graph.calibrate import inference_mode -from brevitas.graph.equalize import activation_equalization_mode from brevitas.graph.quantize import preprocess_for_quantize from brevitas.graph.target.flexml import preprocess_for_flexml_quantize from brevitas_examples.imagenet_classification.ptq.ptq_common import apply_act_equalization