gaohongkui / GlobalPointer_pytorch

全局指针统一处理嵌套与非嵌套NER的Pytorch实现
380 stars 45 forks source link

请教下loss计算时是如何考虑padding部分的呢? #14

Closed tianylin98 closed 1 year ago

tianylin98 commented 2 years ago

batch内序列肯定有padding的,这部分好像在loss func里没有看到有处理的。

gaohongkui commented 1 year ago

序列内 padding 的去除在这里: https://github.com/gaohongkui/GlobalPointer_pytorch/blob/d32f84b423c787d07ad4092ad0a922dc594987fb/models/GlobalPointer.py#L200