Generating an image on a different machine and environment than the one that was trained will result in a completely black image

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

Generating an image on a different machine and environment than the one that was trained will result in a completely black image #96

Open hiroalchem opened 3 years ago

hiroalchem commented 3 years ago

I used Google Colab for training, and the results were satisfactory. However, when I ran the generation on a different local machine using the model from the training results, it produced a completely black image. Is there any way to solve this problem?

lzc2333 commented 7 months ago

I also met this qusetion.May I ask how do you load the model into a different local machine?