From 46391257ff50848efa1aa251ab3f15dc8b7a2d2c Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Sat, 27 Feb 2021 22:14:21 +0800
Subject: Implement Batch Hard triplet loss and soft margin

---
 utils/configuration.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'utils/configuration.py')

diff --git a/utils/configuration.py b/utils/configuration.py
index 435d815..20aec76 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -43,6 +43,7 @@ class ModelHPConfiguration(TypedDict):
     tfa_squeeze_ratio: int
     tfa_num_parts: int
     embedding_dims: int
+    triplet_is_hard: bool
     triplet_margins: tuple[float, float]
 
 
-- 
cgit v1.2.3