From 98c4dcb678e9d98725fc3756eb72ecae33d85326 Mon Sep 17 00:00:00 2001 From: willem640 <23405263+willem640@users.noreply.github.com> Date: Thu, 27 Jun 2024 13:51:18 +0200 Subject: [PATCH] Fix unpack in latin_sequence_labeling.py This might be because of changed behaviour in a new python version. Unpacking the bert output into sequence_outputs and pooled_outputs gave those values the name of the keys, and not the actual tensors --- case_studies/pos_tagging/scripts/latin_sequence_labeling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/case_studies/pos_tagging/scripts/latin_sequence_labeling.py b/case_studies/pos_tagging/scripts/latin_sequence_labeling.py index b067160..10d1e83 100644 --- a/case_studies/pos_tagging/scripts/latin_sequence_labeling.py +++ b/case_studies/pos_tagging/scripts/latin_sequence_labeling.py @@ -111,7 +111,7 @@ def forward(self, input_ids, token_type_ids=None, attention_mask=None, transform if labels is not None: labels = labels.to(device) - sequence_outputs, pooled_outputs = self.bert(input_ids, token_type_ids=None, attention_mask=attention_mask) + sequence_outputs, pooled_outputs = self.bert(input_ids, token_type_ids=None, attention_mask=attention_mask).values() all_layers=sequence_outputs out=torch.matmul(transforms,all_layers)