为什么会有Warning: masked_fill_ received a mask with dtype torch.uint8，导致看不到结果

chenxiaoyouyou / Bert-BiLSTM-CRF-pytorch

使用谷歌预训练bert做字嵌入的BiLSTM-CRF序列标注模型

MIT License

458 stars 100 forks source link

Open yzphxs opened 3 years ago

sixmilesroad commented 3 years ago

我也遇到这个问题了，解决办法貌似是将mask后加上.bool()。但是你用GPU跑过吗

zelin-x commented 3 years ago

在crf中所有与mask有关得输入全变成.bool()

tianzeshu commented 3 years ago

试了很多次，发现将crf.py文件中的所有 “byte()” 替换为 “bool()”就可以了