summaryrefslogtreecommitdiff
path: root/utils/configuration.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils/configuration.py')
-rw-r--r--utils/configuration.py28
1 files changed, 0 insertions, 28 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 5a5bc0c..fff3876 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -33,23 +33,6 @@ class DataloaderConfiguration(TypedDict):
class ModelHPConfiguration(TypedDict):
ae_feature_channels: int
f_a_c_p_dims: tuple[int, int, int]
- hpm_scales: tuple[int, ...]
- hpm_use_avg_pool: bool
- hpm_use_max_pool: bool
- tfa_num_parts: int
- tfa_squeeze_ratio: int
- embedding_dims: tuple[int]
- triplet_is_hard: bool
- triplet_is_mean: bool
- triplet_margins: tuple[float, float]
-
-
-class SubOptimizerHPConfiguration(TypedDict):
- lr: int
- betas: tuple[float, float]
- eps: float
- weight_decay: float
- amsgrad: bool
class OptimizerHPConfiguration(TypedDict):
@@ -58,22 +41,11 @@ class OptimizerHPConfiguration(TypedDict):
eps: float
weight_decay: float
amsgrad: bool
- auto_encoder: SubOptimizerHPConfiguration
- hpm: 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):