From b95ccc3ca132c7403a3fbd57d6ac6c9f07699e43 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 4 Mar 2023 17:19:30 +0000 Subject: [PATCH] save --- aaa_helpers.txt | 6 +++--- configs/iccv/config_abl_all_cape_f.py | 6 +++--- configs/iccv/config_abl_pyr_32_64_128_256.py | 10 +++++----- .../config_abl_pyr_32_64_128_256_skirt.py | 10 +++++----- cvae_pyramid.py | 20 +++++++++++-------- 5 files changed, 28 insertions(+), 24 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 703babd..5ba4c26 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -23,7 +23,7 @@ nohup python cvae.py --name="lhunor_abl_all" > nohup/iccv/ablation/abl_all_3.out nohup python cvae.py --name="lhunor_abl_all_gnn" > nohup/iccv/ablation/abl_all_gnn.out 2>&1 & nohup python cvae.py --name="lhunor_abl_all_gnn_2" > nohup/iccv/ablation/abl_all_gnn.out 2>&1 & -nohup python cvae.py --name="lhunor_abl_all_cape_f" > nohup/iccv/ablation/abl_all_cape_f.out 2>&1 & +nohup python cvae.py --name="lhunor_abl_all_cape_f" > nohup/iccv/ablation/abl_all_cape_f_2.out 2>&1 & nohup python cvae.py --name="lhunor_abl_all_skirt" > nohup/iccv/ablation/abl_all_skirt.out 2>&1 & nohup python cvae.py --name="lhunor_abl_curv" > nohup/iccv/ablation/abl_curv.out 2>&1 & @@ -97,8 +97,8 @@ nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_128_cape_m_00032_SL" > n nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_128_cape_f_03223_SL" > nohup/iccv/pyr/iccv_pyr_64_128_cape_f_03223_SL.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_64_128_cape_f_03223_SS" > nohup/iccv/pyr/iccv_pyr_64_128_cape_f_03223_SS.out 2>&1 & -nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256" > nohup/iccv/pyr/iccv_pyr_32_64_128_256.out 2>&1 & -nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_skirt" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_skirt.out 2>&1 & +nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_2.out 2>&1 & +nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_skirt" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_skirt_2.out 2>&1 & nohup python evaluate_ae_pyramid.py --name evaluate_ae_iccv_pyr_32 > nohup/iccv/evaluate/iccv_pyr_32.out 2>&1 & nohup python evaluate_ae_pyramid.py --name evaluate_ae_iccv_pyr_32_skirt > nohup/iccv/evaluate/iccv_pyr_32_skirt.out 2>&1 & diff --git a/configs/iccv/config_abl_all_cape_f.py b/configs/iccv/config_abl_all_cape_f.py index 6f76c16..0193560 100644 --- a/configs/iccv/config_abl_all_cape_f.py +++ b/configs/iccv/config_abl_all_cape_f.py @@ -18,8 +18,8 @@ config['use_norm_loss'] = True ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False) assert ablation_count == 1 or ablation_count == 0 -config['gpu'] = '6' -config['name'] = 'abl_all_cape_f' +config['gpu'] = '2' +config['name'] = 'abl_all_cape_f_2' config['notes'] = '' config['tags'] = ['iccv', 'abl_all'] @@ -50,7 +50,7 @@ if config['is_cape']: config['latent_dim'] = 128 -config['pretrained_model'] = '' +config['pretrained_model'] = 'checkpoints/iccv/abl_all_cape_f' ############### config['lr'] = 0.0001# 0.0001 diff --git a/configs/iccv/config_abl_pyr_32_64_128_256.py b/configs/iccv/config_abl_pyr_32_64_128_256.py index 4444cd1..85af18a 100644 --- a/configs/iccv/config_abl_pyr_32_64_128_256.py +++ b/configs/iccv/config_abl_pyr_32_64_128_256.py @@ -1,7 +1,7 @@ config = dict() -config['gpu'] = '3' -config['name'] = 'iccv_pyr_32_64_128_256' +config['gpu'] = '0' +config['name'] = 'iccv_pyr_32_64_128_256_2' config['notes'] = '' config['tags'] = ['pyr', 'iccv'] @@ -22,13 +22,13 @@ config['use_offsets'] = False config['use_norm_loss'] = True config['use_curvature'] = True -config['pretrained_model'] = '' +config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_32_64_128_256' config['pretrained_model_iteration'] = '' config['pyramid'] = dict() config['pyramid']['resolutions'] = [32, 64, 128, 256] -config['pyramid']['pretrained_pyramid'] = 'checkpoints/iccv/iccv_pyr_32_64_128_2' -config['pyramid']['pretrained_pyramid_iteration'] = '125000' +config['pyramid']['pretrained_pyramid'] = '' +config['pyramid']['pretrained_pyramid_iteration'] = '' config['pyramid']['final_upscale_enhancer'] = '' config['train_dir'] = '/home/workspace/local-data/train' diff --git a/configs/iccv/config_abl_pyr_32_64_128_256_skirt.py b/configs/iccv/config_abl_pyr_32_64_128_256_skirt.py index 6213a79..0d42f49 100644 --- a/configs/iccv/config_abl_pyr_32_64_128_256_skirt.py +++ b/configs/iccv/config_abl_pyr_32_64_128_256_skirt.py @@ -1,7 +1,7 @@ config = dict() -config['gpu'] = '0' -config['name'] = 'iccv_pyr_32_64_128_256_skirt' +config['gpu'] = '1' +config['name'] = 'iccv_pyr_32_64_128_256_skirt_2' config['notes'] = '' config['tags'] = ['pyr', 'iccv'] @@ -22,13 +22,13 @@ config['use_offsets'] = False config['use_norm_loss'] = True config['use_curvature'] = True -config['pretrained_model'] = '' +config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_32_64_128_256_skirt' config['pretrained_model_iteration'] = '' config['pyramid'] = dict() config['pyramid']['resolutions'] = [32, 64, 128, 256] -config['pyramid']['pretrained_pyramid'] = 'checkpoints/iccv/iccv_pyr_32_64_128_skirt' -config['pyramid']['pretrained_pyramid_iteration'] = '175000' +config['pyramid']['pretrained_pyramid'] = '' +config['pyramid']['pretrained_pyramid_iteration'] = '' config['pyramid']['final_upscale_enhancer'] = '' config['train_dir'] = '/home/workspace/local-data_128/train_skirt' diff --git a/cvae_pyramid.py b/cvae_pyramid.py index ff932e9..06d1e47 100644 --- a/cvae_pyramid.py +++ b/cvae_pyramid.py @@ -1,6 +1,6 @@ import os import sys -from configs.iccv.config_abl_pyr_64_128_cape_f_03223_SS import config +from configs.iccv.config_abl_pyr_32_64_128_256_skirt import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] import tensorflow as tf @@ -803,18 +803,22 @@ if __name__ == '__main__': if config['pretrained_model'] != '': # assert len(config['pyramid']['resolutions']) == 1 load_model(model, config['pretrained_model'], iteration=config['pretrained_model_iteration']) - - if len(config['pyramid']['resolutions']) == 1: model.static_enc.trainable = False model.body_enc.trainable = False if config['use_curvature']: model.curv_enc.trainable = False else: - load_model(model, config['pyramid']['pretrained_pyramid'], iteration=(config['pyramid']['pretrained_pyramid_iteration'] if 'pretrained_pyramid_iteration' in config.keys() else '')) - model.static_enc.trainable = False - model.body_enc.trainable = False - if config['use_curvature']: - model.curv_enc.trainable = False + if len(config['pyramid']['resolutions']) == 1: + model.static_enc.trainable = False + model.body_enc.trainable = False + if config['use_curvature']: + model.curv_enc.trainable = False + else: + load_model(model, config['pyramid']['pretrained_pyramid'], iteration=(config['pyramid']['pretrained_pyramid_iteration'] if 'pretrained_pyramid_iteration' in config.keys() else '')) + model.static_enc.trainable = False + model.body_enc.trainable = False + if config['use_curvature']: + model.curv_enc.trainable = False # load_model(model.levels[0].upscale_enhancer, config['pyramid']['levels'][0]['upscale_enhancer'])