请教关于博客上的一个问题(问答任务 via Bert)
Closed this issue · 1 comments
qijunhui commented
在博客 [问答任务 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]))
qijunhui commented
已解决