From 54e72012a845416fb2db56ac7a77622a954dfa61 Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Tue, 13 Aug 2019 22:14:27 +0300
Subject: [PATCH] AMC: small fixes
---
examples/auto_compression/amc/amc.py | 2 +-
examples/auto_compression/amc/amc_args.py | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/examples/auto_compression/amc/amc.py b/examples/auto_compression/amc/amc.py
index 684277d..5155867 100755
--- a/examples/auto_compression/amc/amc.py
+++ b/examples/auto_compression/amc/amc.py
@@ -56,7 +56,7 @@ class AutoCompressionSampleApp(classifier.ClassifierCompressor):
def train_auto_compressor(self):
using_fm_reconstruction = self.args.amc_prune_method == 'fm-reconstruction'
fixed_subset, sequential = (using_fm_reconstruction, using_fm_reconstruction)
- msglogger.info("ADC: fixed_subset=%s\tsequential=%s" % (fixed_subset, sequential))
+ msglogger.info("AMC: fixed_subset=%s\tsequential=%s" % (fixed_subset, sequential))
train_loader, val_loader, test_loader = classifier.load_data(self.args, fixed_subset, sequential)
self.args.display_confusion = False
diff --git a/examples/auto_compression/amc/amc_args.py b/examples/auto_compression/amc/amc_args.py
index 3d4edd6..bf1ce47 100755
--- a/examples/auto_compression/amc/amc_args.py
+++ b/examples/auto_compression/amc/amc_args.py
@@ -1,4 +1,4 @@
-def add_automl_args(argparser, arch_choices=None, enable_pretrained=False):
+def add_automl_args(argparser):
"""
Helper function to make it easier to add command-line arguments for AMC to any application.
@@ -55,4 +55,4 @@ def add_automl_args(argparser, arch_choices=None, enable_pretrained=False):
help="Number of filters/channels to group")
group.add_argument('--amc-ranking-noise', type=float, default=0.,
help='Strcuture ranking noise')
- return argparser
\ No newline at end of file
+ return argparser
--
GitLab