diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 40535dc..a1ed97d 100644 --- a/aaa_helpers.txt +++ b/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 diff --git a/configs/iccv/config_abl_pyr_32.py b/configs/iccv/config_abl_pyr_32.py index a8d4070..705bf60 100644 --- a/configs/iccv/config_abl_pyr_32.py +++ b/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: diff --git a/configs/iccv/config_abl_pyr_32_skirt.py b/configs/iccv/config_abl_pyr_32_skirt.py index 881cd2c..df4c444 100644 --- a/configs/iccv/config_abl_pyr_32_skirt.py +++ b/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'] diff --git a/cvae_pyramid.py b/cvae_pyramid.py index 4010495..2f094a3 100644 --- a/cvae_pyramid.py +++ b/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