Skip to content
Snippets Groups Projects
Commit 8e3f04cc authored by Neta Zmora's avatar Neta Zmora
Browse files

object detection: remove unsupported summaries

compute-summary and png-summary currently work with image classifiers
only.
parent 8002b13f
No related branches found
No related tags found
No related merge requests found
...@@ -280,7 +280,7 @@ def create_activation_stats_collectors(model, *phases): ...@@ -280,7 +280,7 @@ def create_activation_stats_collectors(model, *phases):
def add_distiller_compression_args(parser): 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 = parser.add_argument_group('Distiller related arguemnts')
distiller_parser.add_argument('--summary', type=lambda s: s.lower(), choices=SUMMARY_CHOICES, action='append', 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)) help='print a summary of the model, and exit - options: | '.join(SUMMARY_CHOICES))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment