New features
- Migration to
pydantic=2.*
(#105) - Supports custom collate function thanks to @fbosshard (#83)
- Adds auto mixed precision to
BaseTrainer
thanks to @liamchalcroft (#90)
Minor changes
- Unifies Gaussian likelihood for all (VAE-based) model implementations (#104)
- Updates
predict
method inRHVAE
thanks to @soumickmj (#80) - Adds clamping to
SVAE
model for stability thanks to @soumickmj (#79)