How to train the discriminator only?

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

How to train the discriminator only? #99

Open zhyyfaan opened 2 years ago

zhyyfaan commented 2 years ago

I want to use the discriminator only for other projects ,without the generator. I want to know how to set the number of epoch and how to evaluate the result of the training. And I want to check if I just use the aux_loss in your code can train it well.