diff --git a/src/batdetect2/cli/train.py b/src/batdetect2/cli/train.py index 64cf28b..8e9f451 100644 --- a/src/batdetect2/cli/train.py +++ b/src/batdetect2/cli/train.py @@ -13,8 +13,8 @@ __all__ = ["train_command"] @cli.command(name="train") @click.argument("train_dataset", type=click.Path(exists=True)) @click.option("--val-dataset", type=click.Path(exists=True)) -@click.option("--model-path", type=click.Path(exists=True)) -@click.option("--targets-config", type=click.Path(exists=True)) +@click.option("--model", "model_path", type=click.Path(exists=True)) +@click.option("--targets", "targets_config", type=click.Path(exists=True)) @click.option("--ckpt-dir", type=click.Path(exists=True)) @click.option("--log-dir", type=click.Path(exists=True)) @click.option("--config", type=click.Path(exists=True))