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

train non-pyramid with raw curv inputs

Browse Source
This commit is contained in:
luni
2023-01-27 22:22:15 +00:00
parent 6e6b0a6604
commit 33232fa945
3 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/config_abl_all_post.py
+4 -3
View File
@@ -19,9 +19,10 @@ config['only_curvatures'] = False
ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False)
assert ablation_count == 1 or ablation_count == 0
config['gpu'] = '1'
config['name'] = 'al_all_post'
config['gpu'] = '3'
config['name'] = 'abl_all_raw_r2'
config['notes'] = ''
config['tags'] = ['review_2']
config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy' #custom_data/train_no_shape_avg_uv.npy # train_no_shape_avg_uv train_no_shape_one_hot_avg_uv
if config['use_skinning']:
@@ -45,7 +46,7 @@ config['test_dir'] = '/home/workspace/data/dataset/test_ready'
config['static_model'] = 'checkpoints/post/static_post_2'
config['body_model'] = 'checkpoints/post/body_post_2'
config['curvature_model'] = 'checkpoints/post/curv_post_1'
config['curvature_model'] = 'checkpoints/post/curv_raw_r2'
config['smpl_res'] = 14475