Network modification

This commit change the auto-encoder by removing fc and optimizing latent space features

1 files changed, 2 insertions, 2 deletions

diff --git a/config.py b/config.py
index 66eab98..04f7b36 100644
--- a/config.py
+++ b/config.py
@@ -11,7 +11,7 @@ config: Configuration = {
         # Recorde disentangled image or not
         'image_log_on': False,
         # The number of subjects for validating (Part of testing set)
-        'val_size': 10,
+        'val_size': 20,
     },
     # Dataset settings
     'dataset': {
@@ -51,7 +51,7 @@ config: Configuration = {
             # Auto-encoder feature channels coefficient
             'ae_feature_channels': 64,
             # Appearance, canonical and pose feature dimensions
-            'f_a_c_p_dims': (192, 192, 96),
+            'f_a_c_p_dims': (192, 192, 128),
             # HPM pyramid scales, of which sum is number of parts
             'hpm_scales': (1, 2, 4, 8),
             # Global pooling method