Expected input batch_size.. to match target batch_size

🤗Transformers
1

Hello everyone,

I have a problem when I use the method ‘.train()’ about a Seq2Seq trainer; it returns me that there is this kind of problem abut the batch_size, how can i fix it?

2 Likes