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

lot more training

Browse Source
This commit is contained in:
luni
2023-03-03 12:45:49 +00:00
parent d93678ee1a
commit 3b414e76d3
25 changed files with 1632 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_abl_pyr_64_cape_f.py
+6 -5
View File
@@ -1,6 +1,6 @@
config = dict()
config['gpu'] = '6'
config['gpu'] = '2'
config['name'] = 'iccv_pyr_64_cape_f'
config['notes'] = ''
config['tags'] = ['pyr', 'iccv', 'cape_f']
@@ -20,8 +20,9 @@ config['remove_global_orientation'] = True
config['use_extra_losses'] = True # edge, collision, 3d mesh
config['use_offsets'] = False
config['use_norm_loss'] = True
config['use_curvature'] = True
config['pretrained_model'] = ''
config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_64_cape_f'
config['pretrained_model_iteration'] = ''
config['pyramid'] = dict()
@@ -29,9 +30,9 @@ config['pyramid']['resolutions'] = [64]
config['pyramid']['pretrained_pyramid'] = ''
config['pyramid']['final_upscale_enhancer'] = ''
config['train_dir'] = '/home/workspace/local-data/cape_processed/train/female'
config['valid_dir'] = '/home/workspace/local-data/cape_processed/test/female'
config['test_dir'] = '/home/workspace/local-data/cape_processed/test/female'
config['train_dir'] = '/home/workspace/local-data_127/cape_processed/train/female'
config['valid_dir'] = '/home/workspace/local-data_127/cape_processed/test/female'
config['test_dir'] = '/home/workspace/local-data_127/cape_processed/test/female'
config['batch_size'] = 4
config['is_skirt'] = False
config['is_cape'] = True