diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-27 22:16:12 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-27 22:16:12 +0800 |
commit | ec863bad4933f25a5bf14fef2541df43c4d9430f (patch) | |
tree | 98f424edba3e394170ea595d868a36d135e9a138 /config.py | |
parent | 66d7b0e6f80aa6feb5e4e0af9af7ba78db641f9e (diff) | |
parent | 46391257ff50848efa1aa251ab3f15dc8b7a2d2c (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.py | 6 |
1 files changed, 4 insertions, 2 deletions
@@ -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 |