v0.6.6

ziatdinovmax released this 11 Mar 02:14

· 396 commits to master since this release

9a9fd7a

New functionality

Added option for controlled information capacity increase to VAE and rVAE (jVAE and jrVAE have them by default). Based on Eq (8) in https://arxiv.org/pdf/1804.03599.pdf

Bug fixes

Fixed bug that was preventing from loading (older) AtomAI models without saved optimizer in their meta-dictionary

Other

Fixed some inconsistencies in classes/methods documentation
Tested against the newest PyTorch release (1.8.0)

Assets 2