解决Bert 报错:AttributeError: ‘str’ object has no attribute ‘softmax’在这里插入图片描述
需要将这里改为:

output= self.bert(input_ids = batch_seqs, attention_mask = batch_seq_masks,token_type_ids=batch_seq_segments, labels = labels)
loss = output.loss
logits = output.logits
Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐