I resolved it the issue,  disc() (fc1) took size [32, 784] and I had the input tensor (adv_ex) as [32,28,28]
This is the current issue that I’m struggling with MmBackward returned an invalid gradient at index 0? 1