From ecd9b139809826b459fbc6487f56c1037bea3b90 Mon Sep 17 00:00:00 2001
From: Guangli Li <guangli.li@hotmail.com>
Date: Thu, 6 Dec 2018 22:56:44 +0800
Subject: [PATCH] Update compress_classifier.py (#97)
Update the examples of earlyexit arguments which were not consistent with descriptions
---
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 a4c9a98..8ff0eb3 100755
--- a/examples/classifier_compression/compress_classifier.py
+++ b/examples/classifier_compression/compress_classifier.py
@@ -151,9 +151,9 @@ parser.add_argument('--adc-params', dest='ADC_params', default=None, help='temp
parser.add_argument('--confusion', dest='display_confusion', default=False, action='store_true',
help='Display the confusion matrix')
parser.add_argument('--earlyexit_lossweights', type=float, nargs='*', dest='earlyexit_lossweights', default=None,
- help='List of loss weights for early exits (e.g. --lossweights 0.1 0.3)')
+ help='List of loss weights for early exits (e.g. --earlyexit_lossweights 0.1 0.3)')
parser.add_argument('--earlyexit_thresholds', type=float, nargs='*', dest='earlyexit_thresholds', default=None,
- help='List of EarlyExit thresholds (e.g. --earlyexit 1.2 0.9)')
+ help='List of EarlyExit thresholds (e.g. --earlyexit_thresholds 1.2 0.9)')
parser.add_argument('--num-best-scores', dest='num_best_scores', default=1, type=int,
help='number of best scores to track and report (default: 1)')
parser.add_argument('--load-serialized', dest='load_serialized', action='store_true', default=False,
--
GitLab