From 6ffc1c06f66277d37877fc13fb1ffa585598d6d7 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 14 Jan 2021 23:43:29 +0800
Subject: Enable optimizer fine tuning
---
utils/configuration.py | 14 ++++++++++++++
1 file changed, 14 insertions(+)
(limited to 'utils')
diff --git a/utils/configuration.py b/utils/configuration.py
index 9a8c2ae..8b265e8 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):
--
cgit v1.2.3