skirt pyramid 32

aaa_helpers.txt

@@ -35,6 +35,7 @@ nohup python upscale_test.py --name="iccv_128_2_cape_m" > nohup/iccv/upscale/icc
 nohup python upscale_test.py --name="iccv_256_2_cape_m" > nohup/iccv/upscale/iccv_256_2_cape_m.out 2>&1 &
 
 nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32" > nohup/iccv/pyr/iccv_pyr_32.out 2>&1 &
+nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_skirt" > nohup/iccv/pyr/iccv_pyr_32_skirt.out 2>&1 &
 
 skirts!!!!
 pyramid next week

configs/iccv/config_abl_pyr_32.py

@@ -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:

configs/iccv/config_abl_pyr_32_skirt.py

@@ -1,6 +1,6 @@
 config = dict()
 
-config['gpu'] = '0'
+config['gpu'] = '5'
 config['name'] = 'iccv_pyr_32_skirt'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv']

cvae_pyramid.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_pyr_32 import config
+from configs.iccv.config_abl_pyr_32_skirt import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf