Hello,
I am having similar issue while running trainer.train():
ValueError: Expected input batch_size (664) to match target batch_size (8).
Checkpoint: bert-base-uncased
Dataset: jmamou/augmented-glue-sst2
Can anyone please help?
Thanks,
2 Likes