Releases: suzuki-2001/pytorch-proVLAE
Releases · suzuki-2001/pytorch-proVLAE
v1.0.0
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)