Skip to content

v0.6.5
Compare
Choose a tag to compare
@ziatdinovmax ziatdinovmax released this 26 Feb 20:01
· 407 commits to master since this release
v0.6.5
7fb6dcd
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

New functionalities:

  • Add VAE that can learn (simultaneously) both discrete and continuous latent representations
    image
  • Add option for annealing of the KL terms associated with rotation and image content to rVAE

Bug fixes

  • Fix bug that prevented rVAE from working in non-square images
  • Fix bug that was causing VAE decoders to "forget" apply sigmoid in evaluation regime after training with BCE with logits

Other improvements

  • Add option to set custom encoder and decoder modules in all VAEs
  • Add a substantial amount of tests for VI trainer and VAE modules
  • Update docs
Assets 2
Loading