diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-28 23:11:05 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-28 23:11:05 +0800 |
| commit | fed5e6a9b35fda8306147e9ce772dfbf3142a061 (patch) | |
| tree | 7ad4ffef229e64699522a468c11f071d5a45a725 /utils/configuration.py | |
| parent | b837336695213e3e660992fcd01c5a52c654ea4f (diff) | |
Implement sum of loss default in [1]
[1]A. Hermans, L. Beyer, and B. Leibe, “In defense of the triplet loss for person re-identification,” arXiv preprint arXiv:1703.07737, 2017.
Diffstat (limited to 'utils/configuration.py')
| -rw-r--r-- | utils/configuration.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/utils/configuration.py b/utils/configuration.py index 20aec76..31eb243 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -44,6 +44,7 @@ class ModelHPConfiguration(TypedDict): tfa_num_parts: int embedding_dims: int triplet_is_hard: bool + triplet_is_mean: bool triplet_margins: tuple[float, float] |