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

save

Browse Source
This commit is contained in:
luni
2023-03-04 20:32:16 +00:00
parent 76cb457a54
commit dfdf8d408e
4 changed files with 17 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_abl_pyr_32_64_128_reg.py
+7 -7
View File
@@ -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
configs/iccv/config_abl_pyr_64_128_256_cape_f.py
+6 -7
View File
@@ -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: