But I got an error which said something like the size of the last batch does not equal to the previous batch sizes. I used torch.utils.data.DataLoader(training_data, batch_size=64, shuffle=True, num_workers=2)