That’s weird, because my test folder also have two subfolders(both are named the same things), and when I print train_data.classes and test_data.classes I get the same two classes for both.
I did figure out where the code gets it number 10 from, because when I changed my batch size to 100, i got “Target size (torch.Size([100])) must be the same as input size (torch.Size([2]))”
Is there a reason why the batch size is influencing the target size?
Thanks again