There is a newer version of the record available.

Published February 26, 2021 | Version v0.6.5
Software Open

pycroscopy/atomai: v0.6.5

  • 1. Codacy

Description

New functionalities:

  • Add VAE that can learn (simultaneously) both discrete and continuous latent representations
  • 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

Files

pycroscopy/atomai-v0.6.5.zip

Files (43.8 MB)

Name Size Download all
md5:1783200bbd33569a7f6f7fcfd678546e
43.8 MB Preview Download

Additional details

Related works