diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-28 23:12:25 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-28 23:12:25 +0800 |
commit | e04f54d0bfc8fc711e53561065d772dae1926b64 (patch) | |
tree | 0996a345a272795b9f77de397fb6aaca6e159088 /config.py | |
parent | ec863bad4933f25a5bf14fef2541df43c4d9430f (diff) | |
parent | fed5e6a9b35fda8306147e9ce772dfbf3142a061 (diff) |
Merge branch 'master' into python3.8
# Conflicts:
# utils/configuration.py
# utils/triplet_loss.py
Diffstat (limited to 'config.py')
-rw-r--r-- | config.py | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -53,7 +53,7 @@ config: Configuration = { # Use 1x1 convolution in dimensionality reduction 'hpm_use_1x1conv': False, # HPM pyramid scales, of which sum is number of parts - 'hpm_scales': (1, 2, 4), + 'hpm_scales': (1, 2, 4, 8), # Global pooling method 'hpm_use_avg_pool': True, 'hpm_use_max_pool': True, @@ -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, }, |