Update lightweight_gan.py

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.63k stars 222 forks source link

Closed CaptainStabs closed 2 years ago

CaptainStabs commented 2 years ago

Fix import per issue #90

CaptainStabs commented 2 years ago

Slightly embarrassing, I thought it had worked but apparently it didn't :(