Thanks, ptrblck, for your reply.
No, I’m working on multiclass classification where there is no overlap between the classes.
In my case, what do you suggest?
Also, I can not use softmax since my loss function is CrossEntropyLoss which has built-in softmax (as I understand from the documentation of pytorch).
Thanks,