Neural Coder mod launcher arg: "strategy" to "approach"
@@ -28,8 +28,8 @@ def parse_args():
parser.add_argument("--opt", type=str, default="",
help="optimization feature to enable")
- parser.add_argument("--strategy", type=str, default="static",
- help="quantization strategy")
+ parser.add_argument("--approach", type=str, default="static",
+ help="quantization approach (strategy)")
parser.add_argument('--config', type=str, default="",
help='quantization configuration file path')
@@ -53,11 +53,11 @@ shutil.copy(args.script, script_copied)
# optimize on copied script with Neural Coder
from neural_coder import enable
if args.opt == "":
- if args.strategy == "static":
+ if args.approach == "static":
features=["pytorch_inc_static_quant_fx"]
- if args.strategy == "static_ipex":
+ if args.approach == "static_ipex":
features=["pytorch_inc_static_quant_ipex"]
- if args.strategy == "dynamic":
+ if args.approach == "dynamic":
features=["pytorch_inc_dynamic_quant"]
else:
features=[args.opt]
