From 6002b2d2017912f90e8917e6e8b71b78ce58e7c2 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Mon, 1 Mar 2021 18:20:38 +0800
Subject: New scheduler and new config

---
 utils/configuration.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

(limited to 'utils/configuration.py')

diff --git a/utils/configuration.py b/utils/configuration.py
index 31eb243..0f8d9ff 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):
-- 
cgit v1.2.3