diff options
Diffstat (limited to 'utils/configuration.py')
-rw-r--r-- | utils/configuration.py | 28 |
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): |