I am trying to implement the gradient penalty part into the WGAN implementation of DeBlurGAN. I check the already existing implementation on gradient penalty and tried adding it