summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-27 22:16:12 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-27 22:16:12 +0800
commitec863bad4933f25a5bf14fef2541df43c4d9430f (patch)
tree98f424edba3e394170ea595d868a36d135e9a138 /config.py
parent66d7b0e6f80aa6feb5e4e0af9af7ba78db641f9e (diff)
parent46391257ff50848efa1aa251ab3f15dc8b7a2d2c (diff)
Merge branch 'master' into python3.8
# Conflicts: # models/model.py # utils/configuration.py # utils/triplet_loss.py
Diffstat (limited to 'config.py')
-rw-r--r--config.py6
1 files changed, 4 insertions, 2 deletions
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