AMC: small fixes

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

examples/auto_compression/amc/amc_args.py

-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