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

Can't find "__main__" module (sorry if noob question) #127

Closed SPotaufeux closed 2 years ago

SPotaufeux commented 2 years ago

Hello, I hope it's not too much of a noob question, I don't have any background in coding.

After creating the env and installing Pytorch I ran "python setup.py install" and then I ran "python lightweight_gan --data /source --image-size 512" (I filled a "source" folder with pictures of fishes) but I get the error "can't find 'main' module". More exactly, C:\Programmes perso\Logiciels\Anaconda\envs\lightweightgan\python.exe: can't find 'main' module in 'C:\Programmes perso\Logiciels\LightweightGan\lightweight_gan' I tried to copy and rename some of the other modules (init, lightweight_gan...), the code seems to start to run but stops before doing anything. So I guess some file must be missing, or did I do something wrong ?

Thanks a lot for the repo and have a nice day

rgaiacs commented 2 years ago

After you install this package (python setup.py install), you should use

lightweight_gan --data /source --image-size 512

instead of

python lightweight_gan --data /source --image-size 512
SPotaufeux commented 2 years ago

That was the issue, thanks !

rgaiacs commented 2 years ago

@SPotaufeux can you close the issue?

SPotaufeux commented 2 years ago

Ah, yes