This dataset is the In-domain part of the training set of VisRAG it includes 122,752 Query-Document (Q-D) Pairs from openly available academic datasets. Our training data is organized with a batch size of 128, ensuring that all data within the same batch comes from the same dataset.

Dataset

ArXivQA 25,856

ChartQA 4,224

MP-DocVQA 10,624

InfoVQA 17,664

PlotQA 56,192

SlideVQA 8,192

from datasets import… See the full description on the dataset page: https://huggingface.co/datasets/openbmb/VisRAG-Ret-Train-In-domain-data.