Skip to content
Snippets Groups Projects
Unverified Commit e04e5824 authored by Neta Zmora's avatar Neta Zmora Committed by GitHub
Browse files

image-classifier: non-functional refactoring

parents f89ba961 c3d6a1c3
No related branches found
No related tags found
No related merge requests found
...@@ -225,7 +225,7 @@ def init_classifier_compression_arg_parser(): ...@@ -225,7 +225,7 @@ def init_classifier_compression_arg_parser():
help='collect activation statistics on phases: train, valid, and/or test' help='collect activation statistics on phases: train, valid, and/or test'
' (WARNING: this slows down training)') ' (WARNING: this slows down training)')
parser.add_argument('--activation-histograms', '--act-hist', parser.add_argument('--activation-histograms', '--act-hist',
type=distiller.utils.float_range_argparse_checker(exc_min=True), type=float_range(exc_min=True),
metavar='PORTION_OF_TEST_SET', metavar='PORTION_OF_TEST_SET',
help='Run the model in evaluation mode on the specified portion of the test dataset and ' help='Run the model in evaluation mode on the specified portion of the test dataset and '
'generate activation histograms. NOTE: This slows down evaluation significantly') 'generate activation histograms. NOTE: This slows down evaluation significantly')
......
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