diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-02-02 09:07:02 +0000 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2023-02-06 11:51:52 +0000 |
commit | c6cfc78d5245c550016b9709686d2b32ab3fcd5b (patch) | |
tree | 42786a3755370d0fefea659d7e532a5483955239 /src/mlia/cli/options.py | |
parent | f1eaff3c9790464bed3183ff76555cf815166f50 (diff) | |
download | mlia-c6cfc78d5245c550016b9709686d2b32ab3fcd5b.tar.gz |
MLIA-461 Add parameter for the output directory
- Add CLI parameter --output-dir
- Rename ExecutionContext property working_dir into output_dir
- Remove logic for default command as it is no longer needed
Change-Id: I6387f6b688520ba1fc69a80167587297353620f6
Diffstat (limited to 'src/mlia/cli/options.py')
-rw-r--r-- | src/mlia/cli/options.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/mlia/cli/options.py b/src/mlia/cli/options.py index e01f107..d154646 100644 --- a/src/mlia/cli/options.py +++ b/src/mlia/cli/options.py @@ -202,6 +202,18 @@ def add_backend_options( ) +def add_output_directory(parser: argparse.ArgumentParser) -> None: + """Add parameter for the output directory.""" + parser.add_argument( + "--output-dir", + type=Path, + help="Path to the directory where MLIA " + "stores artifacts, e.g. logs, target profiles and model files. " + "If not specified then MLIA will use temporary " + "directory instead.", + ) + + def parse_optimization_parameters( pruning: bool = False, clustering: bool = False, |