Hello,
I am curious about this topic as well, not necessarily vision but a classification.
I am using BCELoss() my prediction has shape torch.Size([1,32,1]) and my labels were torch.Size([32]). Which was resulting in a UserWarning: Using a target size (torch.Size([32])) that is different to the input size (torch.Size([1, 32, 1])) is deprecated.
Does this effect the performance of my model? Should I reshape my labels or my predictions to feed into the loss function?
Cheers!