summaryrefslogtreecommitdiff
path: root/utils
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-04-08 12:57:54 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-04-08 12:57:54 +0800
commit543b163c1e8b3914be5c69fa746033da27952449 (patch)
tree0daac884baf8fc70f49d4adce0fdec21660151a5 /utils
parentb54596ab5ce41110100214cd76ff50acc65b589a (diff)
parenta66e5a456a2b036b2b787371da3160efc559031f (diff)
Merge branch 'master' into python3.8
Diffstat (limited to 'utils')
-rw-r--r--utils/configuration.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 8dcae07..f18dab0 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -8,6 +8,7 @@ class SystemConfiguration(TypedDict):
CUDA_VISIBLE_DEVICES: str
save_dir: str
image_log_on: bool
+ image_log_steps: int
val_size: int
@@ -65,11 +66,13 @@ class OptimizerHPConfiguration(TypedDict):
class SubSchedulerHPConfiguration(TypedDict):
start_step: int
+ stop_step: int
final_gamma: float
class SchedulerHPConfiguration(TypedDict):
start_step: int
+ stop_step: int
final_gamma: float
auto_encoder: SubSchedulerHPConfiguration
hpm: SubSchedulerHPConfiguration