Skip to content
forked from vfcosta/coegan

Code used in the experiments of the paper COEGAN: Evaluating the Coevolution Effect in Generative Adversarial Networks http://gecco-2019.sigevo.org

Notifications You must be signed in to change notification settings

ajinkya933/coegan

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

This repository contains the implementation of COEGAN and all code used in the evaluation and comparison with other methods, as presented in the paper COEGAN: Evaluating the Coevolution Effect in Generative Adversarial Networks http://gecco-2019.sigevo.org.

Environment Setup

Install pytorch:

conda install pytorch torchvision cuda90 -c pytorch

Install dependencies:

pip install -r requirements.txt

Run All Unit Tests

python -m unittest discover

Experiments

Configure the Training Setup

Edit the experimental settings in evolution/config.py.

Start the Training:

python ./train.py

Visualize the Results

Run JupyterLab

jupyter lab

Results

See below the results of the experiments presented in the paper:

MNIST

Fid Score

FID Score

Generated Samples

FID Score

Fashion-MNIST

Fid Score

FID Score

Generated Samples

FID Score

Changes to make:

https://stackoverflow.com/questions/55124407/output-and-broadcast-shape-mismatch-in-mnist-torchvision change this line in /evolution/gan_train.py:

transform = transforms.Compose([transforms.ToTensor(),
  transforms.Normalize((0.5,), (0.5,))
])

About

Code used in the experiments of the paper COEGAN: Evaluating the Coevolution Effect in Generative Adversarial Networks http://gecco-2019.sigevo.org

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 90.5%
  • Jupyter Notebook 9.5%