gradient penalty was not actually implemented  grad_penalty should be added to loss_D
gradient penalty was not actually implemented

grad_penalty should be added to loss_D