In this project, we implement a basic GAN model using PyTorch to generate handwritten digits.
To learn more about GANs, check a mathematical tutorial in my github.
mnist.ipynb ----- A comprehensive notebook, step-by-step explanation of model bulidng
mnist.py ----- A fast-run version, script that you can see results within approximate hour after training
utils.py ----- Print resuls, save model and related
- Python 3
- torch
- UC Berkeley IEOR FinTech Capstone Team (Edward Tiong, Fan Xi, Yitong, Yishuang, Yuxin)
- UC Berkeley IEOR FinTech Faculty (Prof Xin Guo, Cheng-Ju)
- more to be added
If you face any issues or have any questions, please contact me at qi_chen@berkeley.edu