skirt pyramid 32

2023-02-26 22:34:22 +00:00
parent 64ecb900f5
commit 9148d63dc3
4 changed files with 4 additions and 3 deletions
@@ -151,7 +151,7 @@ for res in config['pyramid']['resolutions']:
    level['decoder_depths'] = level['encoder_depths']
    level['decoder_dims'] = level['encoder_dims']
    level['avg_garment_path'] = ''
-    level['upscale_enhancer'] = 'checkpoints/iccv/iccv_32_2'#'checkpoints/up_3_d32_64_2'
+    level['upscale_enhancer'] = 'checkpoints/iccv/iccv_32_2_skirt'#'checkpoints/up_3_d32_64_2'
    level['upscale_enhancer_freeze'] = True
    config['pyramid']['levels'].append(level)
  else:
@@ -1,6 +1,6 @@
 config = dict()

-config['gpu'] = '0'
+config['gpu'] = '5'
 config['name'] = 'iccv_pyr_32_skirt'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv']