p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!13539 fix occasional precision failure of CosineEmbeddingLoss 

From: @david-he91
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2021-03-18 19:12:34 +08:00 committed by Gitee
parent 90d1f24b9a b9a773955c
commit 93c21f99e2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/loss/loss.py
View File

@ -908,7 +908,7 @@ class CosineEmbeddingLoss(_Loss):
prod_sum = self.reduce_sum(x1 * x2, (1,)) prod_sum = self.reduce_sum(x1 * x2, (1,))
square1 = self.reduce_sum(F.square(x1), (1,)) square1 = self.reduce_sum(F.square(x1), (1,))
square2 = self.reduce_sum(F.square(x2), (1,)) square2 = self.reduce_sum(F.square(x2), (1,))
denom = F.sqrt(square1 * square2) denom = F.sqrt(square1) * F.sqrt(square2)
cosine = prod_sum / denom cosine = prod_sum / denom
pos_value = 1.0 - cosine pos_value = 1.0 - cosine