I’m experiencing a problem where the adversarial loss (adv_loss) does not decrease during training when using my Flux implementation. I think my d_loss is reducing more faster than expected. Did you come across problem like this ? or Can you give advices to solve this problem ?
(I am using one H100 so my batch size is 2)

I’m experiencing a problem where the adversarial loss (
adv_loss) does not decrease during training when using my Flux implementation. I think my d_loss is reducing more faster than expected. Did you come across problem like this ? or Can you give advices to solve this problem ?(I am using one H100 so my batch size is 2)