gaohongkui / GlobalPointer_pytorch

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

在 train_step 调用loss_fun的时候 第一个参数是pred, 但是定义loss_fun的时候 第一个参数是 label #19

Closed CheungZeeCn closed 1 year ago

CheungZeeCn commented 1 year ago

在 train_step 调用loss_fun的时候 第一个参数是pred

https://github.com/gaohongkui/GlobalPointer_pytorch/blob/d32f84b423c787d07ad4092ad0a922dc594987fb/train.py#L159

但是定义的时候 第一个参数是 label def loss_fun(y_true, y_pred):

计算loss的过程是对称的,所以没有出错,不过还是建议作者改一下?