My PyTorch implementation of https://arxiv.org/abs/2103.01950
Work in progress
Implementations of 3D versions of:
- PixelSNAIL (only for training) https://arxiv.org/pdf/1712.09763.pdf
- VQ-VAE-2 https://arxiv.org/abs/1906.00446
Missing:
- Generating new videos/3D stuff with 3D PixelSNAIL. Currently only ready for training
- Examples
- prob more stuff
Thx to https://github.com/rosinality/vq-vae-2-pytorch and https://github.com/EugenHotaj/pytorch-generative