From 1a3d42f83c2f3c3f67653796a82db14ec74d899d Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Sat, 29 Oct 2022 20:06:34 +0000
Subject: [PATCH] add config cape updates

---
 configs/config_ae_body.py | 4 ++--
 configs/config_ae_curv.py | 9 ++++-----
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/configs/config_ae_body.py b/configs/config_ae_body.py
index ab674dd..5f50b51 100644
--- a/configs/config_ae_body.py
+++ b/configs/config_ae_body.py
@@ -21,10 +21,10 @@ config['batch_size'] = 32
 config['signature_dynamic'] = { 'id': (),
                           'pose_shape': (72 + 11,),
                           'uv_body_posed': (3, 512, 256,),
-                          'mesh_body_posed': (14475, 3,),
+                          'mesh_body_posed': (config['smpl_res'], 3,),
                         }
 config['signature_static'] = { 
-                         'mesh_mask': (14475,),
+                         'mesh_mask': (config['smpl_res'],),
                         }
 
 config['cond_ae']['latent_dim'] = config['latent_dim']
diff --git a/configs/config_ae_curv.py b/configs/config_ae_curv.py
index 452dbaa..b4718b6 100644
--- a/configs/config_ae_curv.py
+++ b/configs/config_ae_curv.py
@@ -2,19 +2,18 @@ from configs.config_base import config
 
 config['gpu'] = '5'
 config['lr'] = 0.001
-config['name'] = 'curv_1_5' #'ae_curv_9'
+config['name'] = 'curv_1' #'ae_curv_9'
 config['notes'] = 'convnext + convnext, 16*8*32, ce loss. flattened latent dim of 128. trained on 100 garments. switched back to sparse CCE, normal one was giving NaN | \
-  try ssim loss | \
-  smaller model | \
+| \
   '
 
 config['cond_ae']['is_label'] = True
 
 config['signature_dynamic'] = { 'id': (),
                           'uv_curvature': (512, 256, 7),
-                          'mesh_curvature':  (14475)
+                          'mesh_curvature':  (config['smpl_res'])
                         }
-config['signature_static'] = { 'mesh_mask': (14475,),
+config['signature_static'] = { 'mesh_mask': (config['smpl_res'],),
                          'uv_mask': (512, 256,),
                         }