From 3df06fb406464c2b0201b3867321552585a5744a Mon Sep 17 00:00:00 2001
From: SunYiran <38311841+Emily0219@users.noreply.github.com>
Date: Tue, 3 Dec 2019 17:07:14 +0800
Subject: [PATCH] Bugfix - Save stats to file when 'qe-calibration' arg is used
(#437)
---
examples/classifier_compression/compress_classifier.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/examples/classifier_compression/compress_classifier.py b/examples/classifier_compression/compress_classifier.py
index 8fe0d4b..a6904f7 100755
--- a/examples/classifier_compression/compress_classifier.py
+++ b/examples/classifier_compression/compress_classifier.py
@@ -99,7 +99,7 @@ def handle_subapps(model, criterion, optimizer, compression_scheduler, pylogger,
args.dataset, add_softmax=True, verbose=False)
do_exit = True
elif args.qe_calibration and not (args.evaluate and args.quantize_eval):
- classifier.acts_quant_stats_collection(model, criterion, pylogger, args)
+ classifier.acts_quant_stats_collection(model, criterion, pylogger, args, save_to_file=True)
do_exit = True
elif args.activation_histograms:
classifier.acts_histogram_collection(model, criterion, pylogger, args)
--
GitLab