1 files changed, 2 insertions, 0 deletions
diff --git a/config.py b/config.py
index 9072982..4c108e2 100644
--- a/config.py
+++ b/config.py
@@ -65,6 +65,8 @@ config: Configuration = {
             'embedding_dims': 256,
             # Batch Hard or Batch All
             'triplet_is_hard': True,
+            # Use non-zero mean or sum
+            'triplet_is_mean': True,
             # Triplet loss margins for HPM and PartNet, None for soft margin
             'triplet_margins': None,
         },