From d12dd6b04a4e7c2b1ee43ab6f36f25d0c35ca364 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Fri, 19 Feb 2021 22:39:49 +0800
Subject: New branch with auto-encoder only

---
 utils/configuration.py | 24 ------------------------
 1 file changed, 24 deletions(-)

(limited to 'utils/configuration.py')

diff --git a/utils/configuration.py b/utils/configuration.py
index 435d815..1b7c8d3 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -32,26 +32,6 @@ class DataloaderConfiguration(TypedDict):
 class ModelHPConfiguration(TypedDict):
     ae_feature_channels: int
     f_a_c_p_dims: tuple[int, int, int]
-    hpm_scales: tuple[int, ...]
-    hpm_use_1x1conv: bool
-    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_margins: tuple[float, float]
-
-
-class SubOptimizerHPConfiguration(TypedDict):
-    lr: int
-    betas: tuple[float, float]
-    eps: float
-    weight_decay: float
-    amsgrad: bool
 
 
 class OptimizerHPConfiguration(TypedDict):
@@ -61,10 +41,6 @@ class OptimizerHPConfiguration(TypedDict):
     eps: float
     weight_decay: float
     amsgrad: bool
-    auto_encoder: SubOptimizerHPConfiguration
-    part_net: SubOptimizerHPConfiguration
-    hpm: SubOptimizerHPConfiguration
-    fc: SubOptimizerHPConfiguration
 
 
 class SchedulerHPConfiguration(TypedDict):
-- 
cgit v1.2.3