Hi Piotr, using sigmoid and then BCEWithLogitsLoss doesn’t work for me. Do you know what could be the problem? Predicted labels stuck at 1 for test set where class 0 is 20% of data - #3 by Mona_Jalal
As you see in that post, I have converted the values to [0, 1] using sigmoid: sig: tensor([0.4937, 0.3869, 0.5705, 0.4727, 0.5873, 0.5160, 0.4642, 0.3229, 0.4283, 0.5563, 0.4485, 0.4954, 0.5997, 0.4649, 0.4181, 0.3966],