pytorch-metric-learning icon indicating copy to clipboard operation
pytorch-metric-learning copied to clipboard

center loss implementation

Open amirhfarzaneh opened this issue 5 years ago • 2 comments

Is center loss considered to be a metric learning approach? If you think it can be part of this repo, I am interested in contributing.

amirhfarzaneh avatar May 26 '20 19:05 amirhfarzaneh

Yes that would be great! I assume you're talking about this paper. In terms of implementation, take a look at NormalizedSoftmaxLoss for an example of how a classification loss is implemented.

KevinMusgrave avatar May 27 '20 08:05 KevinMusgrave

Yes it's the same paper. I will take a look at that example.

amirhfarzaneh avatar May 27 '20 19:05 amirhfarzaneh