scripts/05_finetune.py

"""Phase 4 entrypoint: fine-tune CodeT5+.

Usage:  python scripts/05_finetune.py
Trains on data/processed/train.parquet, saves to data/codet5p-ft/.
"""
import sys
from pathlib import Path

sys.path.append(str(Path(__file__).resolve().parents[1]))
from src.finetune.train_codet5 import finetune

if __name__ == "__main__":
    print("=" * 60, "\nPHASE 4: FINE-TUNE CODET5+\n", "=" * 60, sep="")
    finetune(subset_size=5000, epochs=1)