ValueError: Target size (torch.Size([8])) must be the same as input size (torch.Size([8, 8]))

Beginners
10

Hello setting the labels to int solved the issue for me df["label"] = df["label"].astype(int)

5 Likes