scripts/train_classification.py

from pathlib import Path

import argparse

from thyroid_vfm.config import ClassificationConfig, load_yaml_config
from thyroid_vfm.training.classification import train_classification


def main() -> None:
    parser = argparse.ArgumentParser(description="Train ViT + LoRA on TN5000 classification.")
    parser.add_argument(
        "--config",
        type=Path,
        default=Path("configs/classification_vit_lora.yaml"),
        help="Path to the classification config file.",
    )
    args = parser.parse_args()
    config = load_yaml_config(args.config, ClassificationConfig)
    train_classification(config)


if __name__ == "__main__":
    main()