From 2ef3eeb668d9b8808c770d60d737bd1251b6e680 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Thu, 16 May 2019 12:03:07 +0300
Subject: [PATCH] compress_classifier.py: --summary related fixes
The previous PR merge introduced a couple of small errors when
using the --summary flag.
---
examples/classifier_compression/compress_classifier.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/classifier_compression/compress_classifier.py b/examples/classifier_compression/compress_classifier.py
index cc7965e..883ff92 100755
--- a/examples/classifier_compression/compress_classifier.py
+++ b/examples/classifier_compression/compress_classifier.py
@@ -196,8 +196,8 @@ def main():
# This sample application can be invoked to produce various summary reports.
if args.summary:
- for s in which_summary:
- distiller.model_summary(model, s, dataset)
+ for summary in args.summary:
+ distiller.model_summary(model, summary, args.dataset)
return
if args.export_onnx is not None:
--
GitLab