Skip to content

Welcome to PyTorch GAN Trainer

Quickly train a GAN to generate images!

A simple module for you to directly import and start training different GAN models. You can also log your runs with Weight and Biases.

Aim

  • To create an easy inference to train/save/load different GAN models.
  • To learn how to create a Python package.
  • To learn unit testing with PyTest.
  • To learn how to create documentation with MkDocs.

How to install?

Using pip, run the following command

pip install git+https://github.com/kad99kev/pytorch_gan_trainer.git

How to use?

Please check the Colaboratory notebook to see different examples - Example Notebook

You can also visit the documented example here!

Contributing

  • If you find any bugs, feel free to raise an issue!
  • If you'd like to contribute by adding other GAN architectures, raise a pull request!

Note

Since this project was mainly built as a learning experience, there might be few lingering bugs here and there. In that case, please raise an issue or submit a PR!

License

This project is licensed under the MIT License.