config updates

configs/iccvw/pyr_64_skin_skirt.py

@@ -2,14 +2,14 @@ from utils.convnext_utils import CNBlockConfig
 
 config = dict()
 
-config['gpu'] = '4'
-config['name'] = 'pyr_iccvw_64_skin_skirt'
+config['gpu'] = '0'
+config['name'] = 'pyr_iccvw_64_skin_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'
 
@@ -131,7 +131,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
@@ -157,7 +157,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
@@ -181,7 +181,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