Could you check the length of your Dataset and make sure it’s larger than the batch size of 64?
Since you are using this values for the batch size and drop_last=True, this line of code:
x = next(iter(data))

will raise the exception for datasets with fewer samples.