From 0efe0075a2ffe6cbad31595cab90ecab2c9d6a2d Mon Sep 17 00:00:00 2001 From: Bin Lu Date: Mon, 31 May 2021 14:25:59 +0800 Subject: [PATCH] Update triplet.py --- ppcls/losses/triplet.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ppcls/losses/triplet.py b/ppcls/losses/triplet.py index a8b1a17c..beef5def 100644 --- a/ppcls/losses/triplet.py +++ b/ppcls/losses/triplet.py @@ -141,3 +141,4 @@ class TripletLoss(nn.Layer): y = paddle.ones_like(dist_an) loss = self.ranking_loss(dist_an, dist_ap, y) return {"TripletLoss": loss} + -- GitLab