Hi. Thanks for the reply. How can I go about using NLL loss, considering my labels are a 1 by 2 tensor? Doesn’t NLL loss expect a single value as label i.e. a class. I have 2 classes but I also have data that represents some overlapping between those two classes ([0.5,0.5] labelled data), but I dont wanna label that data as a 3rd class cause its really not.