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

more training

Browse Source
This commit is contained in:
luni
2023-02-27 12:49:18 +00:00
parent b7a8e4dfef
commit c875a6a0ca
7 changed files with 333 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_abl_pyr_32_skirt.py
+6 -6
View File
@@ -1,7 +1,7 @@
config = dict()
config['gpu'] = '5'
config['name'] = 'iccv_pyr_32_skirt'
config['name'] = 'iccv_pyr_32_skirt_2'
config['notes'] = ''
config['tags'] = ['pyr', 'iccv']
@@ -33,7 +33,7 @@ config['train_dir'] = '/home/workspace/data/dataset/local-cache/train_skirt'
config['valid_dir'] = '/home/workspace/local-data/valid_skirt'
config['test_dir'] = '/home/workspace/data/dataset/test_skirt_ready_2'
config['batch_size'] = 4
config['is_skirt'] = False
config['is_skirt'] = True
config['is_cape'] = False
config['dataset_in_memory'] = False
@@ -79,9 +79,9 @@ for res in config['pyramid']['resolutions']:
config['signature_static']['uv_mask' + res_modifier] = (res, res//2)
config['pyramid']['static_model'] = 'checkpoints/iccv/static_iccv_1'
config['pyramid']['body_model'] = 'checkpoints/iccv/body_iccv_1'
config['pyramid']['curvature_model'] = 'checkpoints/iccv/curv_iccv_1'
config['pyramid']['static_model'] = 'checkpoints/iccv/static_iccv_1_skirt'
config['pyramid']['body_model'] = 'checkpoints/iccv/body_iccv_1_skirt'
config['pyramid']['curvature_model'] = 'checkpoints/iccv/curv_iccv_1_skirt'
config['pyramid']['freeze_encoders'] = True
@@ -151,7 +151,7 @@ for res in config['pyramid']['resolutions']:
level['decoder_depths'] = level['encoder_depths']
level['decoder_dims'] = level['encoder_dims']
level['avg_garment_path'] = ''
level['upscale_enhancer'] = 'checkpoints/iccv/iccv_32_2'#'checkpoints/up_3_d32_64_2'
level['upscale_enhancer'] = 'checkpoints/iccv/iccv_32_2_skirt'#'checkpoints/up_3_d32_64_2'
level['upscale_enhancer_freeze'] = True
config['pyramid']['levels'].append(level)
else: