diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 13:59:05 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 13:59:05 +0800 |
| commit | d63b267dd15388dd323d9b8672cdb9461b96c885 (patch) | |
| tree | 5095fc80fb93b946e4cfdee88258ab4fd49a8275 /utils/configuration.py | |
| parent | 08911dcb80ecb769972c2d2659c8ad152bbeb447 (diff) | |
| parent | c74df416b00f837ba051f3947be92f76e7afbd88 (diff) | |
Merge branch 'master' into python3.8
# Conflicts:
# models/hpm.py
# models/rgb_part_net.py
# utils/configuration.py
# utils/triplet_loss.py
Diffstat (limited to 'utils/configuration.py')
| -rw-r--r-- | utils/configuration.py | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/utils/configuration.py b/utils/configuration.py index 376ae0f..8ee08f2 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -33,16 +33,11 @@ class ModelHPConfiguration(TypedDict): ae_feature_channels: int f_a_c_p_dims: Tuple[int, int, int] hpm_scales: Tuple[int, ...] - hpm_use_1x1conv: bool hpm_use_avg_pool: bool hpm_use_max_pool: bool - fpfe_feature_channels: int - fpfe_kernel_sizes: Tuple[Tuple, ...] - fpfe_paddings: Tuple[Tuple, ...] - fpfe_halving: Tuple[int, ...] - tfa_squeeze_ratio: int tfa_num_parts: int - embedding_dims: int + tfa_squeeze_ratio: int + embedding_dims: Tuple[int] triplet_is_hard: bool triplet_is_mean: bool triplet_margins: Tuple[float, float] @@ -63,14 +58,21 @@ class OptimizerHPConfiguration(TypedDict): weight_decay: float amsgrad: bool auto_encoder: SubOptimizerHPConfiguration - part_net: SubOptimizerHPConfiguration hpm: SubOptimizerHPConfiguration - fc: SubOptimizerHPConfiguration + part_net: SubOptimizerHPConfiguration + + +class SubSchedulerHPConfiguration(TypedDict): + start_step: int + final_gamma: float class SchedulerHPConfiguration(TypedDict): start_step: int final_gamma: float + auto_encoder: SubSchedulerHPConfiguration + hpm: SubSchedulerHPConfiguration + part_net: SubSchedulerHPConfiguration class HyperparameterConfiguration(TypedDict): |