diff options
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, |