Refactoring a PyTorch VAE to PyTorch Lightning

 Since we've been focusing on VAE's recently, let's take look at how to refactor one into PyTorch Lightning.



Comments

Popular posts from this blog

Simulating the Universe with Machine Learning

CycleGAN: a GAN architecture for learning unpaired image to image transformations

Pix2Pix: a GAN architecture for image to image transformation