weight test with gnn

This commit is contained in:
2023-02-24 22:14:16 +00:00
parent 9fbd09386c
commit 5062ed95d8
3 changed files with 51 additions and 7 deletions
+2 -2
View File
@@ -20,8 +20,8 @@ ablation_count = [config['use_skinning'], config['remove_global_orientation'], c
assert ablation_count == 1 or ablation_count == 0
config['gpu'] = '4'
config['name'] = 'iccv_weights_test_6'
config['notes'] = 'smaller weight loss coeff, 1e-6, convolutional weight learning, continued from _5'
config['name'] = 'iccv_weights_test_10'
config['notes'] = 'smaller weight loss coeff, 1e-6, convolutional weight learning, add gnn'
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']: