diff --git a/examples/object_detection_compression/compress_detector.py b/examples/object_detection_compression/compress_detector.py index 07718a637722d68974e25c040b5e52152f00c09c..ffd30c46efd743ee0de471780ed5a640630aa64a 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))