BaaL v1.2.1
Changelogs
Features
- Initial support for ensembles. Example to come.
- Initial support for Pytorch Lightning. Example here.
Bugfixes
- Fix BALD for binary classification
- Fix Random heuristic for generators
- Fix
to_cudafor strings. - Fix a bug where MCDropconnect would not work with DataParallel
Misc
- Warning when no layers are affected by
patch_layersin MCDropout, MCDropconnect.