Browse Source

fix occasional precision failure of CosineEmbeddingLoss

tags/v1.2.0-rc1
hedongdong 4 years ago
parent
commit
b9a773955c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/loss/loss.py

+ 1
- 1
mindspore/nn/loss/loss.py View File

@@ -909,7 +909,7 @@ class CosineEmbeddingLoss(_Loss):
prod_sum = self.reduce_sum(x1 * x2, (1,))
square1 = self.reduce_sum(F.square(x1), (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

pos_value = 1.0 - cosine


Loading…
Cancel
Save