Cannot iterate through Pytorch Dataloader . Unequal tensor

Question

I am training a BERT base model on the imdb dataset. However i am unable to iterate throught the Pytorch Dataloader .

Here is the code fully reproducible. Tr