dxzmpk/dxzmpk.github.io

请教关于博客上的一个问题(问答任务 via Bert)

Closed this issue · 1 comments

在博客 [问答任务 via Bert] 小节中参考了你的代码,但是此处输出的start_scores, end_scores是字符串类型,与博客中所说的形状都为torch.Size([1, 14]并不符,不知道是不是transformers库版本的问题,请问怎么解决呢?

我所使用的版本transformers==4.0.1

start_scores, end_scores = model(torch.tensor([input_ids]), token_type_ids=torch.tensor([token_type_ids]))

已解决