Hi,
Thanks for replying so quickly !
In the code, r_real and r_hat_real are passed to the BCE loss. (same for fake).
Dimensions are [128] for r_real (or r_fake) and [128, 1, 1, 1] for  r_hat_real.
I don’t really understand why it’s failing, and how to correct this. Can you help me ?
Thanks