working pyramid pipeline

configs/config_ae_body.py

@@ -1,8 +1,21 @@
 from configs.config_base import config
 
-config['gpu'] = '0'
+config['gpu'] = '1'
 config['lr'] = 0.0001
-config['name'] = 'ae_body_9'
+config['name'] = 'ae_body_10_3'
+
+config['notes'] = 'convnext + convnext, add dense with tanh at the end of reverse convnext,\
+    flattened bottleneck with 128 latent_dim\ remove global orientation\
+    fixed gender issue with deoriented bodies, again, aaand again\
+    trained on 100 garments\
+    remove multiplications with garment uv mask\
+    larger bottleneck, low res output upscaled\
+    larger batch size, 32 instead of 4\
+    '
+
+config['dataset_in_memory'] = False
+config['batch_size'] = 32
+
 
 config['signature_dynamic'] = { 'id': (),
                           'pose_shape': (72 + 11,),
@@ -11,4 +24,12 @@ config['signature_dynamic'] = { 'id': (),
                         }
 config['signature_static'] = { 
                          'mesh_mask': (14475,),
-                        }
+                        }
+
+config['cond_ae']['upscale_output'] = True
+config['cond_ae']['bottleneck_size'] = 512
+config['cond_ae']['encoder_depths'] = [3, 3, 9, 9, 3, 3]
+config['cond_ae']['encoder_dims'] = [32, 64, 128, 256, 256, config['cond_ae']['bottleneck_size']]
+config['cond_ae']['decoder_input'] = (4, 2, config['cond_ae']['bottleneck_size'])
+config['cond_ae']['decoder_depths'] = [3, 3, 9, 3]
+config['cond_ae']['decoder_dims'] = [config['cond_ae']['bottleneck_size'], 128, 64, 16]