GitHub - AntixK/PyTorch-VAE: A Collection of Variational Autoencoders (VAE) in PyTorch.
GAN とは違って、こっちは VAE のリポジトリ。PyTorch Lightning が使われている。Lightning の使い方の参考にもなるかな・・・。
PyTorch-VAE/cvae.py at master · AntixK/PyTorch-VAE · GitHub
おぉ・・・。ConditionalVAE という名前だけあって CGAN と似たようなことをやっている。