BERT_MRC icon indicating copy to clipboard operation
BERT_MRC copied to clipboard

start_outputs和end_outputs维度问题

Open lxgend opened this issue 4 years ago • 1 comments

https://github.com/JavaStudenttwo/BERT_MRC/blob/69b416471004b4c9ad352c2bf94d9f3ff2d3766b/model/bert_mrc.py#L26

请教一下,start_outputs和end_outputs的nn.Linear(config.hidden_size, 2) 中的2,代表什么意思? 我看香农官方的代码,他们又改成了nn.Linear(config.hidden_size, 1) 非常迷惑,烦请解答,谢谢

lxgend avatar Feb 20 '21 10:02 lxgend

@lxgend 不知道你后续理解了没,我最近在看ner的工作,这个维度2是ce loss,维度1是为了bce。

顺便问一下兄弟,官方的代码你复现精度了吗?我感觉跟paper中的结果有差距

Senwang98 avatar Feb 21 '22 12:02 Senwang98