I am trying to adapt this:
https://github.com/melissa135/Convolutional_AutoEncoder 47 which uses MSELoss to using BCE loss for Binary classification without success
Or this: https://github.com/melissa135/Convolutional_AutoEncoder 47
Which does NOT use the labels (but I need the labels)