summaryrefslogtreecommitdiff
path: root/utils
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-03-01 18:21:16 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-03-01 18:21:16 +0800
commit7489bf339e13282b06a78659f8b8fe9d505e82dd (patch)
treeee7982b5c436c34f442bf28aec23baeb13ca6cb3 /utils
parentc3e354de4a48e483866b15858c3f3b4b7e53f660 (diff)
parent6002b2d2017912f90e8917e6e8b71b78ce58e7c2 (diff)
Merge branch 'master' into python3.8
# Conflicts: # models/model.py
Diffstat (limited to 'utils')
-rw-r--r--utils/configuration.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 2959483..376ae0f 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -57,7 +57,6 @@ class SubOptimizerHPConfiguration(TypedDict):
class OptimizerHPConfiguration(TypedDict):
- start_iter: int
lr: int
betas: Tuple[float, float]
eps: float
@@ -70,8 +69,8 @@ class OptimizerHPConfiguration(TypedDict):
class SchedulerHPConfiguration(TypedDict):
- step_size: int
- gamma: float
+ start_step: int
+ final_gamma: float
class HyperparameterConfiguration(TypedDict):