cape_m pyramid working

configs/pyr_64_cape_m.py

@@ -3,7 +3,7 @@ from utils.convnext_utils import CNBlockConfig
 config = dict()
 
 config['gpu'] = '5'
-config['name'] = 'pyr_torch_64_cape_m'
+config['name'] = 'pyr_torch_64_cape_m_2'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv', 'torch']
 config['script'] = 'scripts/cvae_pyramid.py'
@@ -44,7 +44,7 @@ if config['is_cape']:
   config['smpl_res'] = 3638
 
 config['lr'] = 0.0001
-config['starting_balancing_coeff'] = 0.001
+config['starting_balancing_coeff'] = 0.0001
 config['balancing_activation_limit'] = 0.015
 config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
 config['avg_garment_path'] = ''