File size: 657 Bytes
ad34663 | 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 | 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()
|