From db7edf86af56aa76eaaa2b30f3b6568e401c7ecd Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Tue, 12 Jan 2021 11:38:14 +0800
Subject: Remove TypeDict for python 3.7
---
utils/configuration.py | 59 --------------------------------------------------
utils/misc.py | 5 ++---
2 files changed, 2 insertions(+), 62 deletions(-)
delete mode 100644 utils/configuration.py
(limited to 'utils')
diff --git a/utils/configuration.py b/utils/configuration.py
deleted file mode 100644
index 455abe8..0000000
--- a/utils/configuration.py
+++ /dev/null
@@ -1,59 +0,0 @@
-from typing import TypedDict, Optional, Union, Tuple, Dict
-
-from utils.dataset import ClipClasses, ClipConditions, ClipViews
-
-
-class SystemConfiguration(TypedDict):
- disable_acc: bool
- CUDA_VISIBLE_DEVICES: str
- save_dir: str
-
-
-class DatasetConfiguration(TypedDict):
- name: str
- root_dir: str
- train_size: int
- num_sampled_frames: int
- discard_threshold: int
- selector: Optional[Dict[str, Union[ClipClasses, ClipConditions, ClipViews]]]
- num_input_channels: int
- frame_size: Tuple[int, int]
- cache_on: bool
-
-
-class DataloaderConfiguration(TypedDict):
- batch_size: Tuple[int, int]
- num_workers: int
- pin_memory: bool
-
-
-class HyperparameterConfiguration(TypedDict):
- ae_feature_channels: int
- f_a_c_p_dims: Tuple[int, int, int]
- hpm_scales: Tuple[int, ...]
- hpm_use_avg_pool: bool
- hpm_use_max_pool: bool
- fpfe_feature_channels: int
- fpfe_kernel_sizes: Tuple[Tuple, ...]
- fpfe_paddings: Tuple[Tuple, ...]
- fpfe_halving: Tuple[int, ...]
- tfa_squeeze_ratio: int
- tfa_num_parts: int
- embedding_dims: int
- triplet_margin: float
- lr: int
- betas: Tuple[float, float]
-
-
-class ModelConfiguration(TypedDict):
- name: str
- restore_iter: int
- total_iter: int
-
-
-class Configuration(TypedDict):
- system: SystemConfiguration
- dataset: DatasetConfiguration
- dataloader: DataloaderConfiguration
- hyperparameter: HyperparameterConfiguration
- model: ModelConfiguration
diff --git a/utils/misc.py b/utils/misc.py
index b850830..c047618 100644
--- a/utils/misc.py
+++ b/utils/misc.py
@@ -1,9 +1,8 @@
import os
+from typing import Dict
-from utils.configuration import SystemConfiguration
-
-def set_visible_cuda(config: SystemConfiguration):
+def set_visible_cuda(config: Dict):
"""Set environment variable CUDA device(s)"""
CUDA_VISIBLE_DEVICES = config.get('CUDA_VISIBLE_DEVICES', None)
if CUDA_VISIBLE_DEVICES:
--
cgit v1.2.3