From what I read, we don’t have to use a one-hot vector for 2 classes when using BCEWithLogitLoss. We just need to give a single label column as you have done. So, I think the num_classes should still be 1. Like in https://towardsdatascience.com/pytorch-tabular-binary-classification-a0368da5bb89 2