From a229743d98e141339fcce92d9055a523587daa3d Mon Sep 17 00:00:00 2001
From: Neta Zmora <neta.zmora@intel.com>
Date: Mon, 12 Aug 2019 01:27:11 +0300
Subject: [PATCH] AMC args: cleanup add_automl_args arguments, fix
documentation
---
examples/auto_compression/amc/amc_args.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/examples/auto_compression/amc/amc_args.py b/examples/auto_compression/amc/amc_args.py
index 3d4edd6..6688080 100755
--- a/examples/auto_compression/amc/amc_args.py
+++ b/examples/auto_compression/amc/amc_args.py
@@ -1,6 +1,6 @@
-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.
+ Helper function which defines command-line arguments specific to AMC.
Arguments:
argparser (argparse.ArgumentParser): Existing parser to which to add the arguments
--
GitLab