diff --git a/examples/auto_compression/amc/amc.py b/examples/auto_compression/amc/amc.py index 684277d0c0617d049652f6142bcec29829cd049f..5155867c227117f81b8dcbade4f06b636bab89b1 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 3d4edd684eeedc64f253a79b62768c054cd385c9..bf1ce47483bdda8496de36e2cae5a7276614d165 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