обучение Берта нескольким предложениям

У меня есть 2 предложения ["first sentence", "another one"] токенизировал ввод, используя batch_encode_plus(). Во время тренировки появляется следующая ошибка:

RuntimeError: расширенный размер тензора (64) должен соответствовать существующему размеру (2) в не одноэлементном измерении 2. Целевые размеры: [64, 2, 64]. Размеры тензор: [1, 2]

0 ответов

Другие вопросы по тегам