From 8e3f04cc492107586a0f227a7b5ef35a70e76088 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Mon, 2 Dec 2019 18:55:41 +0200
Subject: [PATCH] object detection: remove unsupported summaries
compute-summary and png-summary currently work with image classifiers
only.
---
examples/object_detection_compression/compress_detector.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/examples/object_detection_compression/compress_detector.py b/examples/object_detection_compression/compress_detector.py
index 07718a6..ffd30c4 100755
--- a/examples/object_detection_compression/compress_detector.py
+++ b/examples/object_detection_compression/compress_detector.py
@@ -280,7 +280,7 @@ def create_activation_stats_collectors(model, *phases):
def add_distiller_compression_args(parser):
- SUMMARY_CHOICES = ['sparsity', 'compute', 'model', 'modules', 'png', 'png_w_params']
+ SUMMARY_CHOICES = ['sparsity', 'model', 'modules']
distiller_parser = parser.add_argument_group('Distiller related arguemnts')
distiller_parser.add_argument('--summary', type=lambda s: s.lower(), choices=SUMMARY_CHOICES, action='append',
help='print a summary of the model, and exit - options: | '.join(SUMMARY_CHOICES))
--
GitLab