Skip to content

KyleChen400/GAN_Mnist_toy_project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Generative Adversarial Network using PyTorch

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.

Results

File Structure

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

Environment

  • Python 3
  • torch

Acknowledgement

  • 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

Contact

If you face any issues or have any questions, please contact me at qi_chen@berkeley.edu

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published