Hey Johsnon, thanks for the effort mate but it did not work out. I posted the entire code which on this link for you to check (Using a target size (torch.Size([100, 1])) that is different to the input size (torch.Size([100, 100])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size). Thanks again for the help mate!!!