ValueError: not enough values to unpack (expected 3, got 2) from loss, output, logits = model

few-shot-NER-benchmark / BaselineCode

The 3 baseline methods for few-shot NER tasks

56 stars 11 forks source link

ValueError: not enough values to unpack (expected 3, got 2) from loss, output, logits = model #9

Open sparkfyb opened 6 months ago

sparkfyb commented 6 months ago

When I runloss, output, logits = model(text_1, attention_mask=attention_mask_1, labels=cls_1, dataset = dataset_chosen[j]), I get ValueError: not enough values to unpack (expected 3, got 2), it seem only have loss and output, sorry for such a simple Q, how can I fix it? TY

sparkfyb commented 6 months ago

I think I know it now, that's because BertNER doesn't write unsup in finetune_model, sad:-(