diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-11 21:15:58 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-11 21:15:58 +0800 |
| commit | 7188d71b2b6faf3da527c8d0ade9a32ec4893dc5 (patch) | |
| tree | a012ba1a14bcc4b58833f8ee53cc3114bea5ba0f /utils/triplet_loss.py | |
| parent | d30cf2cb280e83e4a4abe1e9c2abdbba17d903a3 (diff) | |
Implement evaluator
Diffstat (limited to 'utils/triplet_loss.py')
| -rw-r--r-- | utils/triplet_loss.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/utils/triplet_loss.py b/utils/triplet_loss.py index 1899cc9..8c143d6 100644 --- a/utils/triplet_loss.py +++ b/utils/triplet_loss.py @@ -15,8 +15,8 @@ class BatchAllTripletLoss(nn.Module): # Euclidean distance p x n x n x_squared_sum = torch.sum(x ** 2, dim=2) - x1_squared_sum = x_squared_sum.unsqueeze(1) - x2_squared_sum = x_squared_sum.unsqueeze(2) + x1_squared_sum = x_squared_sum.unsqueeze(2) + x2_squared_sum = x_squared_sum.unsqueeze(1) x1_times_x2_sum = x @ x.transpose(1, 2) dist = torch.sqrt( F.relu(x1_squared_sum - 2 * x1_times_x2_sum + x2_squared_sum) |