Solved by SumanthRH in post #6 
              
              
                The PyTorch docs mention the following about the input arguments for nn.CrossEntropyLoss : 
 [image] 
If you have only two classes in your dataset, the loss will still accept it because it expects all target tensor values to be in the range [0, 1000-1]. So there won’t be an error raised. What you’re…