sigmoid activation in last layer of the generator?

AlexiaJM / RelativisticGAN

Code for replication of the paper "The relativistic discriminator: a key element missing from standard GAN"

720 stars 104 forks source link

Closed ahmed-fau closed 5 years ago

ahmed-fau commented 5 years ago

Hi,

Thanks for this interesting work .. just one question regarding the following comments of the code snippet you have provided in the readme page:

No sigmoid activation in last layer of generator because BCEWithLogitsLoss() already adds it

No activation in generator

Is this true? or you mean the discriminator is what should include/exclude sigmoid activation?

Thanks

AlexiaJM commented 5 years ago

You are right, it's the discriminator, not the generator. Fixed it.