Dear Pytorch lightning community,
I am using Pytorch lightning to train a GAN. Thus, for each training_step I have one generator_step and one discriminator_step (similar to https://github.com/PyTorchLightning/pytorch-lightning-bolts/blob/f48357be353b7acdd882379ac3308fbec95dc40d/pl_bolts/models/gans/basic/basic_gan_module.py#L119).
I need to do two steps of the discriminator_step, thus I have to backward two times for each batch. What’s the best way to do it in pytorch lighting?