Implementation of 'lightweight' GAN proposed in ICLR 2021, in Pytorch

Web: https://www.reddit.com/r/datascience/comments/uidwlx/implementation_of_lightweight_gan_proposed_in/

'Lightweight’ GAN is an open-source implementation of ['lightweight' GAN](https://openreview.net/forum?id=1Fqg133qRaI) proposed in ICLR 2021, in Pytorch. The main contributions of the paper is a skip-layer excitation in the generator, paired with autoencoding self-supervised learning in the discriminator. Quoting the one-line summary "converge on single gpu with few hours' training, on 1024 resolution sub-hundred images".

**Install**

`$ pip install lightweight-gan`

**Use**

One command

`$ lightweight_gan --data ./path/to/images --image-size 512`

Model will be saved to `./models/{name}` every 1000 iterations, and samples from the …

2021 datascience gan iclr implementation pytorch