From ca7119e677e14b209b224fafe4de57780113499f Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Sun, 3 Jan 2021 19:00:07 +0800 Subject: Update hyperparameter configuration, implement prototype fit function --- utils/configuration.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) (limited to 'utils/configuration.py') diff --git a/utils/configuration.py b/utils/configuration.py index 965af94..3e98343 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -30,11 +30,21 @@ class DataloaderConfiguration(TypedDict): class HyperparameterConfiguration(TypedDict): - hidden_dim: int + 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] - hard_or_all: str - margin: float class ModelConfiguration(TypedDict): -- cgit v1.2.3