未验证 提交 0efe0075 编写于 作者: B Bin Lu 提交者: GitHub

Update triplet.py

上级 f62604de
...@@ -141,3 +141,4 @@ class TripletLoss(nn.Layer): ...@@ -141,3 +141,4 @@ class TripletLoss(nn.Layer):
y = paddle.ones_like(dist_an) y = paddle.ones_like(dist_an)
loss = self.ranking_loss(dist_an, dist_ap, y) loss = self.ranking_loss(dist_an, dist_ap, y)
return {"TripletLoss": loss} return {"TripletLoss": loss}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册