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

---
 config.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'config.py')

diff --git a/config.py b/config.py
index 03f2f0d..f76cea5 100644
--- a/config.py
+++ b/config.py
@@ -63,8 +63,10 @@ config: Configuration = {
             'tfa_num_parts': 16,
             # Embedding dimension for each part
             'embedding_dims': 256,
-            # Triplet loss margins for HPM and PartNet
-            'triplet_margins': (1.5, 1.5),
+            # Batch Hard or Batch All
+            'triplet_is_hard': True,
+            # Triplet loss margins for HPM and PartNet, None for soft margin
+            'triplet_margins': None,
         },
         'optimizer': {
             # Global parameters
-- 
cgit v1.2.3