From 2b5bc2350df8ebb8e4e5722f5563f2e44823ac34 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Mon, 17 May 2021 17:19:20 +0800 Subject: Model modification Reduce channels in the auto-encoder and add more layers --- config.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'config.py') diff --git a/config.py b/config.py index 726d06e..4f6d248 100644 --- a/config.py +++ b/config.py @@ -51,20 +51,20 @@ config: Configuration = { 'hyperparameter': { 'model': { # Auto-encoder feature channels coefficient - 'ae_feature_channels': 64, + 'ae_feature_channels': 32, # Appearance, canonical and pose feature dimensions - 'f_a_c_p_dims': (192, 192, 128), + 'f_a_c_p_dims': (48, 48, 32), # HPM pyramid scales, of which sum is number of parts 'hpm_scales': (1, 2, 4, 8), # Global pooling method 'hpm_use_avg_pool': True, 'hpm_use_max_pool': True, # Number of parts after Part Net - 'tfa_num_parts': 16, + 'tfa_num_parts': 8, # Attention squeeze ratio 'tfa_squeeze_ratio': 4, # Embedding dimensions for each part - 'embedding_dims': (256, 256), + 'embedding_dims': (32, 64), # Batch Hard or Batch All 'triplet_is_hard': True, # Use non-zero mean or sum -- cgit v1.2.3