Merge branch 'data_parallel' into data_parallel_py3.8

# Conflicts:
#	models/hpm.py
#	models/model.py
#	models/rgb_part_net.py
#	utils/configuration.py
#	utils/triplet_loss.py

1 files changed, 11 insertions, 9 deletions

diff --git a/utils/configuration.py b/utils/configuration.py
index 376ae0f..8ee08f2 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -33,16 +33,11 @@ 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
+    tfa_squeeze_ratio: int
+    embedding_dims: Tuple[int]
     triplet_is_hard: bool
     triplet_is_mean: bool
     triplet_margins: Tuple[float, float]
@@ -63,14 +58,21 @@ class OptimizerHPConfiguration(TypedDict):
     weight_decay: float
     amsgrad: bool
     auto_encoder: SubOptimizerHPConfiguration
-    part_net: SubOptimizerHPConfiguration
     hpm: SubOptimizerHPConfiguration
-    fc: SubOptimizerHPConfiguration
+    part_net: SubOptimizerHPConfiguration
+
+
+class SubSchedulerHPConfiguration(TypedDict):
+    start_step: int
+    final_gamma: float
 
 
 class SchedulerHPConfiguration(TypedDict):
     start_step: int
     final_gamma: float
+    auto_encoder: SubSchedulerHPConfiguration
+    hpm: SubSchedulerHPConfiguration
+    part_net: SubSchedulerHPConfiguration
 
 
 class HyperparameterConfiguration(TypedDict):