barely working decoder

This commit is contained in:
2023-02-24 10:34:53 +00:00
parent c1113379ec
commit fb4e7dcc65
6 changed files with 92 additions and 30 deletions
+5 -5
View File
@@ -19,14 +19,14 @@ 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'] = '3'
config['name'] = 'iccv_gnn_3'
config['gpu'] = '4'
config['name'] = 'iccv_gnn_decoder_1'
config['notes'] = ''
config['tags'] = ['iccv', 'gnn']
config['tags'] = ['iccv', 'gnn', 'ggn_decoder']
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']:
assert (config['use_offsets'] and config['avg_garment_path'] != '') or (not config['use_offsets'] and config['avg_garment_path'] == '')
# if config['use_skinning']:
# assert (config['use_offsets'] and config['avg_garment_path'] != '') or (not config['use_offsets'] and config['avg_garment_path'] == '')
config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision