summaryrefslogtreecommitdiff
path: root/utils
diff options
context:
space:
mode:
Diffstat (limited to 'utils')
-rw-r--r--utils/configuration.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 17796cd..ef6b757 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -43,10 +43,24 @@ class ModelHPConfiguration(TypedDict):
triplet_margin: float
+class SubOptimizerHPConfiguration(TypedDict):
+ lr: int
+ betas: Tuple[float, float]
+ eps: float
+ weight_decay: float
+ amsgrad: bool
+
+
class OptimizerHPConfiguration(TypedDict):
lr: int
betas: Tuple[float, float]
+ eps: float
weight_decay: float
+ amsgrad: bool
+ auto_encoder: SubOptimizerHPConfiguration
+ part_net: SubOptimizerHPConfiguration
+ hpm: SubOptimizerHPConfiguration
+ fc: SubOptimizerHPConfiguration
class SchedulerHPConfiguration(TypedDict):