luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fine tune

Browse Source
This commit is contained in:
luni
2023-01-31 16:31:30 +00:00
parent 6e24c1f41b
commit 2489e28de4
3 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
aaa_helpers.txt
+1
View File
@@ -1,5 +1,6 @@
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3" > nohup/review/pyr_abl_32_64_e2e_r3.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3" > nohup/review/pyr_abl_32_64_e2e_r3.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3_pre" > nohup/review/pyr_abl_32_64_e2e_r3_pre.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3_pre" > nohup/review/pyr_abl_32_64_e2e_r3_pre.out 2>&1 &
nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3_fine" > nohup/review/pyr_abl_32_64_e2e_r3_fine.out 2>&1 &
nohup python evaluate_ae_pyramid.py --name evaluate_pyr_abl_32_64_e2e_r3 > nohup/evaluate/pyr_abl_32_64_e2e_r3.out 2>&1 & nohup python evaluate_ae_pyramid.py --name evaluate_pyr_abl_32_64_e2e_r3 > nohup/evaluate/pyr_abl_32_64_e2e_r3.out 2>&1 &
nohup python calculate_error.py --name error_pyr_abl_32_64_e2e_r3 > nohup/error/pyr_abl_32_64_e2e_r3.out 2>&1 & nohup python calculate_error.py --name error_pyr_abl_32_64_e2e_r3 > nohup/error/pyr_abl_32_64_e2e_r3.out 2>&1 &
configs/config_abl_pyr_32.py
+4 -4
View File
@@ -1,7 +1,7 @@
config = dict() config = dict()
config['gpu'] = '0' config['gpu'] = '0'
config['name'] = 'pyr_abl_32_64_e2e_r3_pre' config['name'] = 'pyr_abl_32_64_e2e_r3_fine_2'
config['notes'] = '' config['notes'] = ''
config['tags'] = ['review_3'] config['tags'] = ['review_3']
@@ -26,7 +26,7 @@ config['pyramid']['resolutions'] = [32, 64]
config['pyramid']['pretrained_pyramid'] = '' config['pyramid']['pretrained_pyramid'] = ''
config['pyramid']['final_upscale_enhancer'] = 'checkpoints/64_512_2' config['pyramid']['final_upscale_enhancer'] = 'checkpoints/64_512_2'
config['pretrained_model'] = 'checkpoints/pyr_abl_32_64_e2e_r3' config['pretrained_model'] = 'checkpoints/pyr_abl_32_64_e2e_r3_fine'
config['pretrained_model_iteration'] = '' config['pretrained_model_iteration'] = ''
config['train_dir'] = '/home/workspace/local-data/train' config['train_dir'] = '/home/workspace/local-data/train'
@@ -44,7 +44,7 @@ config['smpl_res'] = 14475
if config['is_cape']: if config['is_cape']:
config['smpl_res'] = 3638 config['smpl_res'] = 3638
config['lr'] = 0.0001 config['lr'] = 0.00001
config['starting_balancing_coeff'] = 0.0001 config['starting_balancing_coeff'] = 0.0001
config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
config['avg_garment_path'] = '' config['avg_garment_path'] = ''
@@ -68,7 +68,7 @@ config['pyramid']['latent_dim'] = config['latent_dim']
config['pyramid']['add_noise'] = True config['pyramid']['add_noise'] = True
config['pyramid']['noise_amount'] = 0.025 config['pyramid']['noise_amount'] = 0.025
config['pyramid']['loss_weights'] = [1,1,1,1,1] config['pyramid']['loss_weights'] = [2,1,1,1,1]
for res in config['pyramid']['resolutions']: for res in config['pyramid']['resolutions']:
if res != 512: if res != 512:
cvae_pyramid.py
+6 -2
View File
@@ -807,8 +807,12 @@ if __name__ == '__main__':
# assert len(config['pyramid']['resolutions']) == 1 # assert len(config['pyramid']['resolutions']) == 1
load_model(model, config['pretrained_model'], iteration=config['pretrained_model_iteration']) load_model(model, config['pretrained_model'], iteration=config['pretrained_model_iteration'])
model.trainable = False model.static_enc.trainable = False
model.cond_encoder_2.trainable = True model.body_enc.trainable = False
model.curv_enc.trainable = False
# model.trainable = False
# model.cond_encoder_2.trainable = True
# if len(config['pyramid']['resolutions']) == 1: # if len(config['pyramid']['resolutions']) == 1:
# else: # else:
# load_model(model, config['pyramid']['pretrained_pyramid'], iteration=(config['pyramid']['pretrained_pyramid_iteration'] if 'pretrained_pyramid_iteration' in config.keys() else '')) # load_model(model, config['pyramid']['pretrained_pyramid'], iteration=(config['pyramid']['pretrained_pyramid_iteration'] if 'pretrained_pyramid_iteration' in config.keys() else ''))