I’ve modeled my training script on the information in the finetuning with custom datasets documentation ( https://huggingface.co/transformers/custom_datasets.html ). I have both a custom dataset and a custom model (I used the run_language_modeling.py script to pretrain the roberta-base model with ou…

ValueError: Expected input batch_size (16) to match target batch_size (64)

laurb October 16, 2020, 10:12am 2

Tried this using roberta-base as the model as well, and get the same error.

Topic		Replies	Views
ValueError: Expected input batch_size (4096) to match target batch_size (8) Beginners	3	8441	April 2, 2023
ValueError: Expected input batch_size (8) to match target batch_size (280) Beginners	1	1968	November 18, 2024
ValueError: Target size (torch.Size([8])) must be the same as input size (torch.Size([8, 8])) Beginners	15	12611	January 9, 2025
Multilabel sequence classification with Roberta value error expected input batch size to match target batch size 🤗Transformers	1	4242	March 2, 2021
Expected input batch_size (2048) to match target batch_size (4) Beginners	3	1608	May 23, 2022