File size: 751 Bytes
feba2ad |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
#!/usr/bin/env python3
"""
A minimal script to train the Pico language model. In practice, you should just use the
`poetry run train` command to run the training pipeline. Doing so will invoke this script.
Training logic is located in `src/training/trainer.py`.
"""
from pathlib import Path
import click
from src.training.trainer import Trainer
@click.command()
@click.option(
"--config_path",
"config_path",
type=click.Path(exists=True, path_type=Path),
help="Path to the training configuration file",
)
def main(config_path: Path) -> None:
"""Train the Pico language model using the specified configuration."""
trainer = Trainer(config_path=str(config_path))
trainer.train()
if __name__ == "__main__":
main()
|