config updates

configs/iccvw/pyr_64_reg_skirt.py

@@ -2,14 +2,14 @@ from utils.convnext_utils import CNBlockConfig
 
 config = dict()
 
-config['gpu'] = '3'
-config['name'] = 'pyr_iccvw_64_reg_skirt_2'
+config['gpu'] = '4'
+config['name'] = 'pyr_iccvw_64_reg_skirt_3'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv', 'torch']
 config['script'] = 'scripts/cvae_pyramid.py'
 
 config['batch_size'] = 4
-config['is_skirt'] = False
+config['is_skirt'] = True
 config['is_cape'] = False
 config['device'] = 'cuda'
 
@@ -19,7 +19,7 @@ config['use_extra_losses'] =          False # edge, collision, 3d mesh
 config['use_norm_loss'] =             True
 config['use_normals'] =               True
 
-config['pretrained_model'] = 'checkpoints/iccvw/pyr_iccvw_64_reg_skirt'
+config['pretrained_model'] = ''
 config['pretrained_model_iteration'] = ''
 
 config['pyramid'] = dict()
@@ -128,7 +128,7 @@ for res in config['pyramid']['resolutions']:
     level = dict()
     level['resolution'] = 256
     level['kernel_size'] = 5
-    level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_256'
+    level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_256_skirt'
     level['upscale_enhancer_freeze'] = True
     level['bottleneck_size'] = 64 * config['bottleneck_scale']
     level['input_features'] = 3+3+3+3 if config['use_normals'] else 3+3+3
@@ -154,7 +154,7 @@ for res in config['pyramid']['resolutions']:
     level = dict()
     level['resolution'] = 128
     level['kernel_size'] = 5
-    level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_128'
+    level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_128_skirt'
     level['upscale_enhancer_freeze'] = True
     level['bottleneck_size'] = 64 * config['bottleneck_scale']
     level['input_features'] = 3+3+3+3 if config['use_normals'] else 3+3+3
@@ -178,7 +178,7 @@ for res in config['pyramid']['resolutions']:
     level = dict()
     level['resolution'] = 64
     level['kernel_size'] = 3
-    level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_64'
+    level['upscale_enhancer'] = 'checkpoints/iccvw/iccv_64_skirt'
     level['upscale_enhancer_freeze'] = True
     level['bottleneck_size'] = 64 * config['bottleneck_scale']
     level['input_features'] = 3+3+3+3 if config['use_normals'] else 3+3+3