From dfdf8d408e880231c7e2133518d5de17bcc45eae Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 4 Mar 2023 20:32:16 +0000 Subject: [PATCH] save --- aaa_helpers.txt | 4 +++- configs/iccv/config_abl_pyr_32_64_128_reg.py | 14 +++++++------- configs/iccv/config_abl_pyr_64_128_256_cape_f.py | 13 ++++++------- cvae_pyramid.py | 2 +- 4 files changed, 17 insertions(+), 16 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 11da9a8..cf9599b 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -87,12 +87,14 @@ nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_cape_f_03223_SS" > nohup nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_128_cape_f" > nohup/iccv/pyr/iccv_pyr_64_128_cape_f.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_128_cape_m" > nohup/iccv/pyr/iccv_pyr_64_128_cape_m.out 2>&1 & +nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_128_256_cape_f" > nohup/iccv/pyr/iccv_pyr_64_128_256_cape_f.out 2>&1 & + nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_pre_cape_f" > nohup/iccv/pyr/iccv_pyr_64_pre_cape_f.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128" > nohup/iccv/pyr/iccv_pyr_32_64_128.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_skirt" > nohup/iccv/pyr/iccv_pyr_32_64_128_skirt.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_skin" > nohup/iccv/pyr/iccv_pyr_32_64_128_skin_2.out 2>&1 & -nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_reg" > nohup/iccv/pyr/iccv_pyr_32_64_128_reg.out 2>&1 & +nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_reg" > nohup/iccv/pyr/iccv_pyr_32_64_128_reg_2.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_curv" > nohup/iccv/pyr/iccv_pyr_32_64_128_curv.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_128_cape_m_00032_SL" > nohup/iccv/pyr/iccv_pyr_64_128_cape_m_00032_SL.out 2>&1 & diff --git a/configs/iccv/config_abl_pyr_32_64_128_reg.py b/configs/iccv/config_abl_pyr_32_64_128_reg.py index 0efedd4..05b70b2 100644 --- a/configs/iccv/config_abl_pyr_32_64_128_reg.py +++ b/configs/iccv/config_abl_pyr_32_64_128_reg.py @@ -1,7 +1,7 @@ config = dict() -config['gpu'] = '4' -config['name'] = 'iccv_pyr_32_64_128_reg' +config['gpu'] = '1' +config['name'] = 'iccv_pyr_32_64_128_reg_2' config['notes'] = '' config['tags'] = ['pyr', 'iccv', 'reg'] @@ -22,17 +22,17 @@ config['use_offsets'] = False config['use_norm_loss'] = False config['use_curvature'] = True -config['pretrained_model'] = '' +config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_32_64_128_reg' config['pretrained_model_iteration'] = '' config['pyramid'] = dict() config['pyramid']['resolutions'] = [32, 64, 128] -config['pyramid']['pretrained_pyramid'] = 'checkpoints/iccv/iccv_pyr_32_64_reg' -config['pyramid']['pretrained_pyramid_iteration'] = '120000' +config['pyramid']['pretrained_pyramid'] = '' +config['pyramid']['pretrained_pyramid_iteration'] = '' config['pyramid']['final_upscale_enhancer'] = '' -config['train_dir'] = '/home/workspace/local-data/train' -config['valid_dir'] = '/home/workspace/local-data/valid' +config['train_dir'] = '/home/workspace/local-data_127/train' +config['valid_dir'] = '/home/workspace/local-data_127/valid' config['test_dir'] = '/home/workspace/local-data_127/test_ready' config['batch_size'] = 4 config['is_skirt'] = False diff --git a/configs/iccv/config_abl_pyr_64_128_256_cape_f.py b/configs/iccv/config_abl_pyr_64_128_256_cape_f.py index c86473c..6bc4673 100644 --- a/configs/iccv/config_abl_pyr_64_128_256_cape_f.py +++ b/configs/iccv/config_abl_pyr_64_128_256_cape_f.py @@ -1,6 +1,6 @@ config = dict() -config['gpu'] = '3' +config['gpu'] = '4' config['name'] = 'iccv_pyr_64_128_256_cape_f' config['notes'] = '' config['tags'] = ['pyr', 'iccv', 'cape_f'] @@ -107,13 +107,12 @@ for res in config['pyramid']['resolutions']: level['resolution'] = 256 level['latent_dim'] = 64 * config['bottleneck_scale'] level['kernel_size'] = 5 - level['encoder_depths'] = [9, 9, 9, 9] - level['encoder_dims'] = [32, 64, 128, level['latent_dim']] - level['decoder_input'] = (8, 4, 128) - level['decoder_depths'] = [9, 9, 9, 9] - level['decoder_dims'] = [256, 128, 64, 16] + level['encoder_depths'] = [3, 3, 9, 3] + level['encoder_dims'] = [64, 128, 128, level['latent_dim']] + level['decoder_depths'] = level['encoder_depths'] + level['decoder_dims'] = level['encoder_dims'] level['avg_garment_path'] = '' - level['upscale_enhancer'] = 'checkpoints/iccv/iccv_'#'checkpoints/up_3_d256_512_2' + level['upscale_enhancer'] = 'checkpoints/iccv/iccv_256_2_cape_f'#'checkpoints/up_3_d256_512_2' level['upscale_enhancer_freeze'] = True config['pyramid']['levels'].append(level) elif res == 128: diff --git a/cvae_pyramid.py b/cvae_pyramid.py index 8856cd3..7dec104 100644 --- a/cvae_pyramid.py +++ b/cvae_pyramid.py @@ -1,6 +1,6 @@ import os import sys -from configs.iccv.config_abl_pyr_32_64_128_skin import config +from configs.iccv.config_abl_pyr_32_64_128_reg import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] import tensorflow as tf