Skip to content

Pytorch implementation of DCGAN along with an introduction to Fretchet Inception Distance(FID) that serves a better metric to evaluate GANs

Notifications You must be signed in to change notification settings

Ibtastic/GAN-in-Pytorch-with-FID

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

GAN-in-Pytorch-with-FID

The main idea behind this repo is to implement DCGAN in Pytorch with some improvement techniques and implement Fretchet Inception Distance along with it. GANs are one of my favorite neural networks and one of the biggest pain points was assesing it's performance as precisely as we can do for other neural networks. Knowing when to stop the training(reaching Nash equilbirium) or comparing two GAN models was never straightforward. This drawback is overcomed by using Fretchet Inception Distance which is also considered superior to it's predecessor Inception Score.

FID was introduced in the paper GANs Trained by a Two Time-Scale Update Rule Converge to a Local Nash Equilibrium

I have picked up the underlying DCGAN implementation from this Pytorch tutorial and have iteratively improved upon it by some hacks discussed in the notebook and compared performance between experiments using FID(for which I also provide a comprehensive explanation).

About

Pytorch implementation of DCGAN along with an introduction to Fretchet Inception Distance(FID) that serves a better metric to evaluate GANs

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published