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)