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

gnn with weights

Browse Source
This commit is contained in:
luni
2023-02-25 12:33:18 +00:00
parent 5062ed95d8
commit d8337bfcfe
4 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/config_abl_all_post.py
+4 -4
View File
@@ -19,9 +19,9 @@ 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'] = '4'
config['name'] = 'iccv_weights_test_10'
config['notes'] = 'smaller weight loss coeff, 1e-6, convolutional weight learning, add gnn'
config['gpu'] = '5'
config['name'] = 'iccv_weights_test_11'
config['notes'] = 'smaller weight loss coeff, 1e-6, convolutional weight learning, add gnn but smaller and remove softmac from weights convnext'
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']:
@@ -53,7 +53,7 @@ if config['is_cape']:
config['smpl_res'] = 3638
config['pretrained_model'] = 'checkpoints/post/iccv_weights_test_5'
config['pretrained_model'] = ''
###############
config['lr'] = 0.0001# 0.0001