Pytorch Generation Models

Generation models using Pytorch.
I'm going to add new generation models every weekday.
Starting with the basic models!

2021/10/14 After InfoGAN, it will be updated in form of notebook(.ipynb).

Environments

Pytorch : 1.6.0
Torchvision : 0.7.0
torchsummary : 1.5.1
numpy : 1.19.1

Training and Test

python train.py --logdir='temp' --gpu=0 --seed=0 --epochs=30
python test.py --logdir='temp' --gpu=0 --seed=0

Next tasks

1. Add StyleGAN series
   

2. Need New generation models.(like flow based)
   

3. Korean to English
   

4. New dataset

Reference

• Papers for each model
• Hwalsuklee/tensorflow-generative-model-collections