Skip to content

Releases: suzuki-2001/pytorch-proVLAE

v1.0.0

21 Nov 09:56
091fa22
Compare
Choose a tag to compare

PyTorch implementaion of pro-VLAE

  • Various dataset auto download
  • PyTorch distributed data parallel (DDP) training
  • Traverse each ladder's latent variables
  • Visualize reconstruction images
  • Wandb tracking and logging
  • KL annealing (linear, sigmoid, cosine)