Victarry/Image-Generation-models
Generative models (GAN, VAE, Diffusion Models, Autoregressive Models) implemented with Pytorch, Pytorch_lightning and hydra.
Modular architecture decouples models, networks, and datasets through YAML configuration files, enabling rapid experimentation with grid search and multi-experiment execution via Hydra. Implements 15+ canonical architectures across GANs (WGAN, InfoGAN, BiGAN), VAEs (VQ-VAE, Beta-VAE, FactorVAE), diffusion models (DDPM), and autoregressive models (PixelCNN) with PyTorch Lightning for distributed training. Provides pre-trained results on MNIST, CIFAR-10, and CelebA, with composable command-line configuration for hyperparameter sweeps (`model.lr=1e-3,5e-4,1e-4`) and batch experiment execution.
No commits in the last 6 months.
Stars
61
Forks
10
Language
Python
License
—
Category
Last pushed
May 05, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/Victarry/Image-Generation-models"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
yao-jason/MLDS2018SPRING
Machine Learning and having it Deep and Structured (MLDS) in 2018 spring
ivarunsharma/GenaipythonExamples
A progressive, hands-on collection of Generative AI examples in Python
thtang/ADLxMLDS2017
Deep learning works for ADLxMLDS (CSIE 5431) in NTU
adamz799/Paints
Auto paint sketch(from lineart to anime illustration)
musikisomorphie/swd
unsupervised video and image generation