ecg-gan

Check out this wandb run to get a feel for the baseline model or run python train.py to reproduce the results

Suggested reading

• DCGAN TUTORIAL on PyTorch website.
• Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks by Alec Radford, Luke Metz, Soumith Chintala.
• Tips and tricks to make GANs work
  • Focused on points: 6, 7. Applied most of them.

Code Style

This repository uses pre-commit hooks with forced python formatting (black, flake8, and isort):

pip install pre-commit
pre-commit install

Whenever you execute git commit the files altered / added within the commit will be checked and corrected. black and isort can modify files locally - if that happens you have to git add them again. You might also be prompted to introduce some fixes manually.

To run the hooks against all files without running git commit:

pre-commit run --all-files