Skip to content

Add SN-GAN#23

Open
AlekseyKorshuk wants to merge 9 commits intohuggingface:mainfrom
AlekseyKorshuk:sngan
Open

Add SN-GAN#23
AlekseyKorshuk wants to merge 9 commits intohuggingface:mainfrom
AlekseyKorshuk:sngan

Conversation

@AlekseyKorshuk
Copy link
Copy Markdown
Contributor

@AlekseyKorshuk AlekseyKorshuk commented Apr 9, 2022

This implementation based on this paper, DCGAN and crypto-panks.

Model image size upscaled in comparison with other implementations of this GAN 🤗

Solved the problem of transparency with:

  1. Using RGBA
  2. Penalizing the model for transparency

Training script supports:

  • Accelerate
  • Push to hub
  • Weights and Biases
  • Hub datasets
  • All parameters from other training scripts from HugGAN

Copy link
Copy Markdown
Contributor

@nateraw nateraw left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Wow thanks a lot for adding this!! 🚀 Will get a chance to run it a little later on tonight, but just wanted to drop a few comments when I first saw it.

Comment thread huggan/pytorch/sngan/README.md Outdated
Comment thread huggan/pytorch/sngan/train.py
Comment thread huggan/pytorch/sngan/train.py Outdated
@AlekseyKorshuk
Copy link
Copy Markdown
Contributor Author

Updated 👍

@AlekseyKorshuk AlekseyKorshuk requested a review from nateraw April 9, 2022 16:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants