finetuning-scheduler

finetuning-scheduler is a PyTorch Lightning extension for foundation model experimentation with flexible fine-tuning schedules. It supports explicit or regex-based parameter selection and implicit schedules for initial model exploration, plus automatic restoration of best per-phase checkpoints using early-stopping criteria for each phase. An experimental API, FTSController, brings the same schedules to a plain PyTorch training loop.

GitHub repo Directory import
0 shouts Added Sep 23, 2026 github.com
github.com/speediedan/finetuning-scheduler
finetuning-scheduler website preview

Platforms

Links

Tags