I see. Thanks!
So I’ll try to add a Sigmoid layer at the end (to get a probability of belonging to class 1, which then I can compare it to my ground truth class labels with values [0, 1]), and I’ll train using a BCELoss.