search

eriklindernoren / PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.
MIT License
16.22k stars 4.05k forks source link

Bug in ACGAN implementation #141

Open hmrishavbandy opened 3 years ago

hmrishavbandy commented 3 years ago

https://github.com/eriklindernoren/PyTorch-GAN/blob/a163b82beff3d01688d8315a3fd39080400e7c01/implementations/acgan/acgan.py#L100

Later on we use Cross Entropy with this output--however, we need raw logits for cross entropy loss...Maybe change it to NLL loss?