mfai v2.0.0
Dependencies changes
- pytorch >= 2.4 -> v2.5
- timm >= 1.0.11
API changes
- NamedTensor: remove
bare_tensorparameter from functions and change default return type. PR#49 - Model registry:
mfai.torch.models.nn_architectures: [ModelType, list[ModelABC]. PR#50 - SegmentationLightningModule: moved to
mfai.torch.lightning_module. PR#62 - SegmentationLightningModule: test metrics managed with a pytorch MetricCollection. PR#68
- Tokenizers:
MiniTokenizerbecomesMiniGPT2Tokenizerclass andbase_tokenizerbecomesgpt2_tokenizer. PR#35 - UNet: Classes with the word "unet" in their name have been standardized to be spelled "UNet". PR#72
CSINeighborhood: fixed typo in class name. PR#73
Features