mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add ref for emlloss

pull/1886/head
HydrogenSulfate 2022-04-29 13:53:21 +08:00
parent 645a125c40
commit 92cdbe6fb9
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppcls/loss/emlloss.py
View File

@ -23,6 +23,11 @@ from .comfunc import rerange_index
class EmlLoss(paddle.nn.Layer):
"""Ensemble Metric Learning Loss
paper: [Large Scale Strongly Supervised Ensemble Metric Learning, with Applications to Face Verification and Retrieval](https://arxiv.org/pdf/1212.6094.pdf)
code reference: https://github.com/PaddlePaddle/models/blob/develop/PaddleCV/metric_learning/losses/emlloss.py
"""
def __init__(self, batch_size=40, samples_each_class=2):
super(EmlLoss, self).__init__()
assert (batch_size % samples_each_class == 0)