From e6fb9cbd9cf495b92dd725d7c94e97c717d956ea Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 8 Jul 2021 15:15:21 -0500
Subject: [PATCH] Fixed a problem with quantization

---
 hpvm/projects/distiller                    | 2 +-
 hpvm/test/epoch_dnn/torch_dnn/quantizer.py | 2 --
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/hpvm/projects/distiller b/hpvm/projects/distiller
index 30c50f368d..e369794020 160000
--- a/hpvm/projects/distiller
+++ b/hpvm/projects/distiller
@@ -1 +1 @@
-Subproject commit 30c50f368d219efa87d927c0ad0a563bcfa29f0a
+Subproject commit e369794020902decaf50739cd25e8f7250f0d720
diff --git a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
index 7bbbbda45c..05b2a2535b 100644
--- a/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
+++ b/hpvm/test/epoch_dnn/torch_dnn/quantizer.py
@@ -13,8 +13,6 @@ from distiller.quantization import PostTrainLinearQuantizer
 from torch import nn
 from torch.utils.data import DataLoader
 
-from .datasets import CIFAR
-
 PathLike = Union[str, Path]
 STATS_FILENAME = "acts_quantization_stats.yaml"
 QUANT_FILENAME = "layer_quant_params.yaml"
-- 
GitLab