WGAN-Gp Gradient Penalty Computation bug

lkhphuc / Anomaly-XRay-GANs

Anomaly detection on X-Ray MURA dataset using GAN.

48 stars 19 forks source link

Open mostafaelaraby opened 5 years ago

mostafaelaraby commented 5 years ago

In this line of the code you are calling the discrimnator with a single input which is the interpolated image but it also requires another parameter which is the z https://github.com/lkhphuc/Anomaly-XRay-GANs/blob/bbcea0dd8c19d6458a10ba723484d489fd9bfa3b/trainer.py#L157

lkhphuc commented 5 years ago

Thank for the head up. I don't use WGAN-gp often so that if is forgotten. Do you mind in helping to fix that?

mostafaelaraby commented 5 years ago

Sure i will try to fix that but it seems tricky to apply wgan-gp with encoder generator setup