bugfix in train CLI

This commit is contained in:
svlandeg 2020-07-31 14:18:27 +02:00
parent 0b23594953
commit 08d3c36c20
1 changed files with 1 additions and 2 deletions

View File

@ -51,7 +51,7 @@ def train_cli(
referenced in the config.
"""
util.set_env_log(verbose)
verify_cli_args(train_path, dev_path, config_path)
verify_cli_args(train_path, dev_path, config_path, output_path)
overrides = parse_config_overrides(ctx.args)
import_code(code_path)
train(
@ -173,7 +173,6 @@ def train(
progress = tqdm.tqdm(total=training["eval_frequency"], leave=False)
except Exception as e:
if output_path is not None:
raise e
msg.warn(
f"Aborting and saving the final best model. "
f"Encountered exception: {str(e)}",