Open flexibility2 opened 5 years ago
Hi, thanks for your source code. But I notice in the line: https://github.com/KaiyangZhou/pytorch-center-loss/blob/master/center_loss.py#L47 you wrote the center loss as the mean of dis, however, in the paper, the center loss is just the sum of the dis, so could you explain the reason? Thanks a lot
Same confusion with you.
jenfung
commented
4 years ago jenfung
commented
4 years ago
Hi, thanks for your source code. But I notice in the line: https://github.com/KaiyangZhou/pytorch-center-loss/blob/master/center_loss.py#L47 you wrote the center loss as the mean of dis, however, in the paper, the center loss is just the sum of the dis, so could you explain the reason? Thanks a lot