ValueError: аргументы «target» и «output» должны иметь одинаковую форму. Получено: target.shape=(Нет, 512), output.shape ⇐ Python

[Anonymous]

Я пытался обучить модель bert для решения проблемы мультиклассификации:

• Я получил эту ошибку при запуске кода ниже

Целевой и выходной аргументы должны иметь одинаковую форму. Получено: target.shape=(Нет, 512), output.shape=(Нет, 3)

Код: Выделить всё

import tensorflow as tf

epochs = 4

train_dataloader = train_dataset.shuffle(buffer_size=10000).batch(batch_size)
validation_dataloader = val_dataset.batch(batch_size)

# start training
history = model.fit(
train_dataloader,  # train_data
validation_data=validation_dataloader,  # validation_data
epochs=epochs,
verbose=1
)
# save the model
model.save("bert_model.h5")

• Это тест

Код: Выделить всё

for batch in train_dataloader.take(1):
input_ids, attention_masks, labels = batch
print("Batch input_ids shape:", input_ids.shape)
print("Batch attention_masks shape:", attention_masks.shape)
print("Batch labels shape:", labels.shape)

# I got this output
Batch input_ids shape: (16, 512)
Batch attention_masks shape: (16, 512)
Batch labels shape: (16,)

Форму тензора я уже проверил.
Надеюсь на ответ!


Подробнее здесь: https://stackoverflow.com/questions/784 ... e-received