Skip to content

Implementing a GAN using PyTorch to model a polynomial function

Notifications You must be signed in to change notification settings

shivakanthsujit/GAN-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Implementation of GANs using PyTorch

The Generative Adversarial Network implemented in this project was based on the original paper Generative Adversarial Networks by Goodfellow et al.

It was trained to learn a simple polynomial function

ReLU and Leaky ReLU with Dropout were used in the generator and discriminator respectively along with having one-sided label smoothing.

These optimisations were based on the recommendations from Improved Techniques for Training GANs and Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks.

To run this project

pip install -r requirements.txt
python GAN.py

Output of the Network before and after training

Graph of the losses of the Generator and Discriminator

About

Implementing a GAN using PyTorch to model a polynomial function

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published