Skip to content

Latest commit

 

History

History
9 lines (4 loc) · 1.06 KB

README.md

File metadata and controls

9 lines (4 loc) · 1.06 KB

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).