После использования DataLoader из pytorch на больших данных (1k) я получаю эту ошибку, но на меньшем количестве (200) я ее не получаю.
train_loader = DataLoader(dataset=data_train,
batch_size=50,
shuffle=True,
num_workers=2)
Я использую вложения текста.
RuntimeError: Caught RuntimeError in DataLoader worker process 1.
Original Traceback (most recent call last):
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 7 and 8 in dimension 1 at /pytorch/aten/src/TH/generic/THTensor.cpp:612
Почему я получаю это и как этого избежать?