summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-27 22:16:32 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-27 22:16:32 +0800
commit10b1299eb46343e119bc413ba8774dcb1a68595e (patch)
tree4d6d7d3650fbabed5cb66ae15d9e098e07252a9c /config.py
parentaf4d28333b8204d6ab74738a8d3b6ae869cb9344 (diff)
parentec863bad4933f25a5bf14fef2541df43c4d9430f (diff)
Merge branch 'python3.8' into python3.7
# Conflicts: # utils/configuration.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 a7d5371..282a9b4 100644
--- a/config.py
+++ b/config.py
@@ -61,8 +61,10 @@ config = {
'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