Skip to content

v0.1.5

Compare
Choose a tag to compare
@warner-benjamin warner-benjamin released this 16 Oct 04:27
· 24 commits to main since this release
  • Adds support for training Hugging Face Transformers models via HuggingFaceCallback & HuggingFaceLoader
  • Profiler Callbacks updated to work with HuggingFaceCallback & HuggingFaceLoader
  • Compile Callback improvements
    • Support LR Finder
    • Support PyTorch 2.1 Compile format
    • Improve saved model compatibility
  • GradientAccumulation & GradientAccumulationSchedule improvements
    • Auto-detect micro-batch size if using fastai dataloader
    • Drop last macro-batch to match full size training
  • Add DataLoaderMixin for adding fastai dataloader features to non-fastai dataloaders
    • Switch FFCV Loader to use DataLoaderMixin

Full Changelog: v0.1.4...v0.1.5