diff options
Diffstat (limited to 'config.py')
| -rw-r--r-- | config.py | 26 |
1 files changed, 2 insertions, 24 deletions
@@ -7,9 +7,9 @@ config: Configuration = { # GPU(s) used in training or testing if available 'CUDA_VISIBLE_DEVICES': '0', # Directory used in training or testing for temporary storage - 'save_dir': 'runs', + 'save_dir': 'runs/dis_only', # Recorde disentangled image or not - 'image_log_on': False, + 'image_log_on': True, # The number of subjects for validating (Part of testing set) 'val_size': 10, }, @@ -52,23 +52,6 @@ config: Configuration = { 'ae_feature_channels': 64, # Appearance, canonical and pose feature dimensions 'f_a_c_p_dims': (192, 192, 96), - # 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, - # Attention squeeze ratio - 'tfa_squeeze_ratio': 4, - # Embedding dimensions for each part - 'embedding_dims': (256, 256), - # Batch Hard or Batch All - 'triplet_is_hard': True, - # Use non-zero mean or sum - 'triplet_is_mean': True, - # Triplet loss margins for HPM and PartNet, None for soft margin - 'triplet_margins': None, }, 'optimizer': { # Global parameters @@ -83,11 +66,6 @@ config: Configuration = { 'weight_decay': 0.001, # Use AMSGrad or not # 'amsgrad': False, - - # Local parameters (override global ones) - # 'auto_encoder': { - # 'weight_decay': 0.001 - # }, }, 'scheduler': { # Step start to decay |