summaryrefslogtreecommitdiff
path: root/utils/configuration.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-19 22:43:17 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-19 22:43:17 +0800
commit4049566103a00aa6d5a0b1f73569bdc5435714ca (patch)
treed84604773f05eab030ff2106c43cb2c091b6e8fc /utils/configuration.py
parentd12dd6b04a4e7c2b1ee43ab6f36f25d0c35ca364 (diff)
parent969030864495e7c2b419400fd81ee0fad83de41e (diff)
Merge branch 'python3.8' into disentangling_only_py3.8
# Conflicts: # models/hpm.py # models/layers.py # models/model.py # models/part_net.py # models/rgb_part_net.py # utils/configuration.py
Diffstat (limited to 'utils/configuration.py')
-rw-r--r--utils/configuration.py14
1 files changed, 7 insertions, 7 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 1b7c8d3..340815b 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -1,4 +1,4 @@
-from typing import TypedDict, Optional, Union
+from typing import TypedDict, Optional, Union, Tuple, Dict
from utils.dataset import ClipClasses, ClipConditions, ClipViews
@@ -17,14 +17,14 @@ class DatasetConfiguration(TypedDict):
num_sampled_frames: int
truncate_threshold: int
discard_threshold: int
- selector: Optional[dict[str, Union[ClipClasses, ClipConditions, ClipViews]]]
+ selector: Optional[Dict[str, Union[ClipClasses, ClipConditions, ClipViews]]]
num_input_channels: int
- frame_size: tuple[int, int]
+ frame_size: Tuple[int, int]
cache_on: bool
class DataloaderConfiguration(TypedDict):
- batch_size: tuple[int, int]
+ batch_size: Tuple[int, int]
num_workers: int
pin_memory: bool
@@ -37,7 +37,7 @@ class ModelHPConfiguration(TypedDict):
class OptimizerHPConfiguration(TypedDict):
start_iter: int
lr: int
- betas: tuple[float, float]
+ betas: Tuple[float, float]
eps: float
weight_decay: float
amsgrad: bool
@@ -58,8 +58,8 @@ class ModelConfiguration(TypedDict):
name: str
restore_iter: int
total_iter: int
- restore_iters: tuple[int, ...]
- total_iters: tuple[int, ...]
+ restore_iters: Tuple[int, ...]
+ total_iters: Tuple[int, ...]
class Configuration(TypedDict):