summaryrefslogtreecommitdiff
path: root/utils/configuration.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2020-12-27 11:15:51 +0800
committerJordan Gong <jordan.gong@protonmail.com>2020-12-27 11:15:51 +0800
commit81b12f9357ca737599fbc11c732020666ad0873f (patch)
tree21beabf4ad44a0098478e143d8032bb42e5e92c0 /utils/configuration.py
parent31b20fbff0786c998c54b8585de759d02f41eda7 (diff)
Adopt type hinting generics in standard collections (PEP 585)
Diffstat (limited to 'utils/configuration.py')
-rw-r--r--utils/configuration.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/utils/configuration.py b/utils/configuration.py
index 32b9bec..84bd064 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -1,4 +1,4 @@
-from typing import TypedDict, Tuple
+from typing import TypedDict
import torch
@@ -16,12 +16,12 @@ class DatasetConfiguration(TypedDict):
num_sampled_frames: int
discard_threshold: int
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
@@ -29,7 +29,7 @@ class DataloaderConfiguration(TypedDict):
class HyperparameterConfiguration(TypedDict):
hidden_dim: int
lr: int
- betas: Tuple[float, float]
+ betas: tuple[float, float]
hard_or_all: str
margin: float