diff --git a/ppcls/losses/triplet.py b/ppcls/losses/triplet.py index a8b1a17c750f4da4f8c99973443e83d4a08663bb..beef5def0f4a981df9e5baf1c1832a3bde2899f8 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} +