在 train_step 调用loss_fun的时候 第一个参数是pred, 但是定义loss_fun的时候 第一个参数是 label
CheungZeeCn opened this issue · 0 comments
CheungZeeCn commented
在 train_step 调用loss_fun的时候 第一个参数是pred
GlobalPointer_pytorch/train.py
Line 159 in d32f84b
但是定义的时候 第一个参数是 label
def loss_fun(y_true, y_pred):
计算loss的过程是对称的,所以没有出错,不过还是建议作者改一下?