conda install numpy requests pillow matplotlib tqdm -c conda-forge
// GPU cuda 12.1
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
- VAE [code]
- PixelCNN [code]
- GatedPixelCNN [code]
- VQ-VAE [code] [EMA version code]
conda install numpy requests pillow matplotlib tqdm -c conda-forge
// GPU cuda 12.1
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia