summaryrefslogtreecommitdiff
path: root/utils
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-01-21 23:44:34 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-01-21 23:44:34 +0800
commit0b76205ecef02dd62ef2fbc8e12d9389b7cf7868 (patch)
tree71927f5efe2dc3228f49326a89e2536785aa2eb4 /utils
parent8572f5c8292e5798912ad54764c9d3a99afb49ec (diff)
parent04c9d3210ff659bbe00dedb2d193a748e7a97b54 (diff)
Merge branch 'master' into python3.8
# Conflicts: # utils/configuration.py
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):