lucidrains / lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two
MIT License
1.62k stars 220 forks source link

Using Lightweight GAN as a library #135

Closed demiahmed closed 2 years ago

demiahmed commented 2 years ago

I have been using this as a CLI solution for training images.

However would like to know if this library can be used as imports within a python environment (like a jupyter notebook).

Basically would like to use the pytorch models generated from Lightweight GAN training process and perform save, load and image generation as an alternative to the comomand line.

Thanks