Skip to content

v0.4.3

Choose a tag to compare

View all tags
@ChrisNick92 ChrisNick92 released this 06 Apr 04:58
· 16 commits to main since this release
v0.4.3
d5413cc
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

What's New

MPS Support

  • Added DeviceName type alias ("cuda", "mps", "cpu")
  • Trainer and Evaluator now accept an explicit device parameter — users on Apple Silicon can pass device="mps" to leverage the Metal GPU backend
  • Fixed hardcoded "cuda" in PaSST mel spectrogram autocast guard — replaced with x.device.type

DataLoader Efficiency

  • pin_memory is now conditional on CUDA — avoids unsupported behavior on MPS and CPU
  • Added persistent_workers=True — eliminates worker process respawn overhead at epoch boundaries
  • pad_collate_fn now uses torch.stack for equal-length batches (e.g. when segment_duration is set), falling back to pad_sequence for variable-length batches
Assets 2
Loading