just the target? or also the output?
That goes back to the point that we need to have output of shape NxC (C is number of classes) for CEL but for BNE it should be just N, am i understanding correct?
Also, BCEWithLogitsLoss can deal with unbalance data, is there any version of CEL that has that features as well?
Also, Can you please help me understand the way that you define the targets. im confuse on a couple of things about them.