Upload folder using huggingface_hub

feba2ad verified about 1 year ago

751 Bytes

#!/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()

	#!/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()