configs

configs/abl_all.py

@@ -2,12 +2,13 @@ from utils.convnext_utils import CNBlockConfig
 
 config = dict()
 
-config['gpu'] = '0'
-config['name'] = 'abl_torch_all'
+config['gpu'] = '6'
+config['name'] = 'abl_torch_all_2'
 config['notes'] = ''
 config['tags'] = ['iccv', 'abl_all', 'torch']
 config['script'] = 'scripts/cvae.py'
 
+config['pretrained_model'] = 'checkpoints/iccv/abl_torch_all'
 
 config['use_skinning'] =              True # and deshaping
 config['remove_global_orientation'] = True
@@ -43,15 +44,12 @@ config['static_model'] = 'checkpoints/iccv/static_torch_orig_2'
 config['body_model'] = 'checkpoints/iccv/body_torch_1'
 config['curvature_model'] = 'checkpoints/iccv/curv_torch_1'
 
-
 config['smpl_res'] = 14475
 if config['is_cape']:
   config['smpl_res'] = 3638
 
 config['latent_dim'] = 128
 
-config['pretrained_model'] = ''
-
 ###############    
 config['lr'] = 0.0001#                 0.0001
 ###############