Блог

Главная - Вопросы по программированию - Ошибка выполнения: стек ожидает, что каждый тензор будет одинакового размера, но получил [5391] при записи 0 и [8441] при записи 1

Ошибка выполнения: стек ожидает, что каждый тензор будет одинакового размера, но получил [5391] при записи 0 и [8441] при записи 1

#nlp #data-science #multilingual #bert-language-model #question-answering

Вопрос:

не могу понять, где я сейчас ошибаюсь в ответе на этот вопрос во время обучения модели.

код для этого присутствует здесь: https://www.kaggle.com/soumochatterjee/questionansweringwithbert?scriptVersionId=72595700

Может ли кто-нибудь помочь решить эту проблему

выборка данных получена :

 df_train = pd.read_csv(config.TRAINING_FILE)

train_dataset = ChaiiDataset(
        context=df_train.context.values,
        question=df_train.question.values,
        answer=df_train.answer_text.values
    )

idx = 1000

print(train_dataset[idx]['ids'])
print(train_dataset[idx]['mask'])
print(train_dataset[idx]['token_type_ids'])
print(train_dataset[idx]['targets_start'])
print(train_dataset[idx]['targets_end'])
print(train_dataset[idx]['orig_tweet'])
print(train_dataset[idx]['orig_selected'])
print(train_dataset[idx]['question'])
print(train_dataset[idx]['offsets'])
Метки: , ,