Skip to content

pertained models

Compare
Choose a tag to compare
@williamFalcon williamFalcon released this 13 Sep 14:27
· 539 commits to master since this release
634d49a

[0.2.1] - 2020-09-13

Added

  • Added pretrained VAE with resnet encoders and decoders
  • Added pretrained AE with resnet encoders and decoders
  • Added CPC pretrained on CIFAR10 and STL10
  • Verified BYOL implementation

Changed

  • Dropped all dependencies except PyTorch Lightning and PyTorch
  • Decoupled datamodules from GAN (#206)
  • Modularize AE & VAE (#196)

Fixed

  • Fixed gym (#221)
  • Fix L1/L2 regularization (#216)
  • Fix max_depth recursion crash in AsynchronousLoader (#191)