但CRF层想要的mask是[True, False]的矩阵,这样数据类型就不一致,产生了bug。 如果小伙伴们在用pytorch复现CRF时,出现了loss NaN的问题,可以检查一下mask是否工作正常,此文章可以帮助排雷。
https://github.com/sgrvinod/a-PyTorch-Tutorial-to-Sequence-Labeling 有代码 https://zhuanlan.zhihu.com/p/61227299 借鉴或者说copy上面的链接 野博客两三个: https://createmomo.github.io/2017/11/11/CRF-Layer-on-the-Top-of-BiLSTM-5/ 这个讲的非常详细,很好。 https://zhuanlan.zhihu.com/p/273382...