gaohongkui/GlobalPointer_pytorch

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

CheungZeeCn opened this issue · 0 comments

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

loss = criterion(logits, batch_labels)

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

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