Skip to content
This repository has been archived by the owner on Oct 25, 2021. It is now read-only.

catalyst-team/gan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Catalyst logo

Accelerated DL & RL

Build Status CodeFactor Pipi version Docs PyPI Status

Twitter Telegram Slack Github contributors

PyTorch framework for Deep Learning research and development. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing. Being able to research/develop something new, rather than write another regular train loop.
Break the cycle - use the Catalyst!

Project manifest. Part of PyTorch Ecosystem. Part of Catalyst Ecosystem:

  • Alchemy - Experiments logging & visualization
  • Catalyst - Accelerated Deep Learning Research and Development
  • Reaction - Convenient Deep Learning models serving

Catalyst at AI Landscape.


Catalyst.Gan [WIP] Github contributors

Note: this repo uses advanced Catalyst Config API and could be a bit out-of-day right now. Use Catalyst's minimal examples section for a starting point and up-to-day use cases, please.

You will learn how to train your GAN using the Catalyst framework. The main advantage is to customize your experiments in the yaml config instead of the code.

Installation

pip install -r requirements.txt

Run examples

MNIST

# (Goodfellow et. al., 2014: https://arxiv.org/pdf/1406.2661.pdf)
catalyst-dl run -C examples/mnist/configs/vanilla_gan.yml
# (Arjovsky et. al., 2017: https://arxiv.org/abs/1701.07875)
catalyst-dl run -C examples/mnist/configs/wasserstein_gan.yml
# (Gulrahani et. al., 2017: https://arxiv.org/abs/1704.00028)
catalyst-dl run -C examples/mnist/configs/wasserstein_gan_gp.yml
# (Mirza and Osindero, 2014: https://arxiv.org/abs/1411.1784)
catalyst-dl run -C examples/mnist/configs/conditional_gan.yml

Advanced [under construction]

If you want to try right now run from console

(you should download FFHQ dataset before that and specify path in examples/advanced/tconfigs/data/FFHQ.yml)

./examples/advanced/experiments_setup/run.sh

Releases

No releases published

Packages

No packages published

Languages