CODEWITHSUNDEEP
pythonpytorchgenerative-adversarial-networkmathematical-expressions
Najeh NAFTI
Najeh NAFTI

Reputation: 1

Implement mathematical equation for Loss function in pytorch

I implemented a GAN model and I would like the discriminator (UNET Discriminator) to compute decision over all pixels and [D(x)]i,j refer to the discriminator decision at pixel (i; j). The original loss in the paper is based on a vanilla Loss, but I would like to use a WGAN-GP loss in my case. How can I implement the below equation in pytorch

enter image description here

Upvotes: 0

Views: 132

Answers (0)

Related Questions