summaryrefslogtreecommitdiff
path: root/config.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-28 23:12:25 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-28 23:12:25 +0800
commite04f54d0bfc8fc711e53561065d772dae1926b64 (patch)
tree0996a345a272795b9f77de397fb6aaca6e159088 /config.py
parentec863bad4933f25a5bf14fef2541df43c4d9430f (diff)
parentfed5e6a9b35fda8306147e9ce772dfbf3142a061 (diff)
Merge branch 'master' into python3.8
# Conflicts: # utils/configuration.py # utils/triplet_loss.py
Diffstat (limited to 'config.py')
-rw-r--r--config.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/config.py b/config.py
index f76cea5..4c108e2 100644
--- a/config.py
+++ b/config.py
@@ -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,
},