[ML] ValueError: Using a target size (torch.Size([50])) that is different to the input size (torch.Size([50, 1])) is deprecated. Please ensure they have the same size.

 

labels = labels.unsqueeze(1)

추가하여 해결