AttributeError: 'str' object has no attribute 'size'

[PEFT2026]

您好，在我使用"python train_demo.py --encoder roberta --pretrain_ckpt /model/roberta-base"命令时，运行到proto.py的37行出现了"AttributeError: 'str' object has no attribute 'size'“错误，我进行了debug，然后发现proto.py中的第34行support_emb的值是一个str类型，但是却在37行使用了.size属性，为什么会出现这种情况呢，self.sentence_encorder(support)返回的是一个字符类型吗？我该怎么处理呢，谢谢

[luckinoffice]

您好，您这个解决了吗

[shivanouri]

This arises due to the newer versions of Transformers.

In the class BERTSentenceEncoder, in the forward pass, instead of
_, x = self.bert(inputs['word'], attention_mask=inputs['mask'])
You'll have to set:

outputs = self.bert(inputs['word'], attention_mask=inputs['mask'])
x = outputs.pooler_output