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

create end2end laplacian pyramid

Browse Source
This commit is contained in:
luni
2023-01-26 20:39:54 +00:00
parent 8c0436cea8
commit 0dfe7f7659
5 changed files with 133 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/config_abl_pyr_32.py
+8 -4
View File
@@ -1,8 +1,9 @@
config = dict()
config['gpu'] = '0'
config['name'] = 'pyr_abl_32_r1'
config['name'] = 'pyr_abl_32_64_e2e_r3'
config['notes'] = ''
config['tags'] = ['review_3']
# DON'T FORGET TO RESET SSIM LOSS USAGE
# DON'T FORGET TO CHECK MODEL LOADING
@@ -21,9 +22,12 @@ config['use_offsets'] = False
config['pyramid'] = dict()
config['pyramid']['resolutions'] = [32]
config['pyramid']['resolutions'] = [32, 64]
config['pyramid']['pretrained_pyramid'] = ''
config['pyramid']['final_upscale_enhancer'] = 'checkpoints/32_512_4'
config['pyramid']['final_upscale_enhancer'] = 'checkpoints/64_512_2'
config['pretrained_model'] = ''
config['pretrained_model_iteration'] = ''
config['train_dir'] = '/home/workspace/local-data/train'
config['valid_dir'] = '/home/workspace/local-data/valid'
@@ -149,7 +153,7 @@ for res in config['pyramid']['resolutions']:
level['decoder_depths'] = [9, 9]
level['decoder_dims'] = [256, 256]
level['avg_garment_path'] = ''
level['upscale_enhancer'] = ''#'checkpoints/up_3_d32_64_2'
level['upscale_enhancer'] = 'checkpoints/32_512_4'#'checkpoints/up_3_d32_64_2'
level['upscale_enhancer_freeze'] = True
config['pyramid']['levels'].append(level)
else: