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

convolutional weights model

Browse Source
This commit is contained in:
luni
2023-02-17 13:11:35 +00:00
parent 61a609c8f7
commit 33eff16c8c
5 changed files with 50 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/config_abl_all_post.py
+4 -3
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'] = '5'
config['name'] = 'iccv_weights_test_3'
config['notes'] = ''
config['name'] = 'iccv_weights_test_5'
config['notes'] = 'smaller weight loss coeff, 1e-6, convolutional weight learning'
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']:
@@ -93,7 +93,8 @@ config['cvae']['encoder_depths'] = [3, 3, 3, 9, 3]
config['cvae']['encoder_dims'] = [32, 64, 128, 256, 64]
config['cvae']['decoder_depths'] = config['cvae']['encoder_depths']
config['cvae']['decoder_dims'] = config['cvae']['encoder_dims']
config['cvae']['decoder_weights_depths'] = [3, 3, 3, 5, 3]
config['cvae']['decoder_weights_dims'] = [24, 32, 64, 128, 256]
config['cond_ae_static'] = dict()
config['cond_ae_static']['is_label'] = False