From db7edf86af56aa76eaaa2b30f3b6568e401c7ecd Mon Sep 17 00:00:00 2001 From: Jordan Gong 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 From 2fa77264fb5002683da902577acea6451e4c2942 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 20:39:39 +0800 Subject: Delete empty files --- utils/configuration.py | 0 1 file changed, 0 insertions(+), 0 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 e69de29..0000000 -- cgit v1.2.3