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
configs/config_abl_pyr_32.py
+4 -4
View File
@@ -1,7 +1,7 @@
config = dict()
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['tags'] = ['review_3']
@@ -26,7 +26,7 @@ config['pyramid']['resolutions'] = [32, 64]
config['pyramid']['pretrained_pyramid'] = ''
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['train_dir'] = '/home/workspace/local-data/train'
@@ -44,7 +44,7 @@ config['smpl_res'] = 14475
if config['is_cape']:
config['smpl_res'] = 3638
config['lr'] = 0.0001
config['lr'] = 0.00001
config['starting_balancing_coeff'] = 0.0001
config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
config['avg_garment_path'] = ''
@@ -68,7 +68,7 @@ config['pyramid']['latent_dim'] = config['latent_dim']
config['pyramid']['add_noise'] = True
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']:
if res != 512: