@@ -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'] = '5'
|
||||
config['name'] = 'iccv_weights_test_8'
|
||||
config['notes'] = 'smaller weight loss coeff, 1e-6, convolutional weight learning, changed convnext reverse'
|
||||
config['gpu'] = '4'
|
||||
config['name'] = 'iccv_weights_test_6'
|
||||
config['notes'] = 'smaller weight loss coeff, 1e-6, convolutional weight learning, continued from _5'
|
||||
|
||||
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'] = ''
|
||||
config['pretrained_model'] = 'checkpoints/post/iccv_weights_test_5'
|
||||
|
||||
###############
|
||||
config['lr'] = 0.0001# 0.0001
|
||||
@@ -91,9 +91,9 @@ config['cvae']['noise_amount'] = 0.00001
|
||||
config['cvae']['kernel_size'] = 7
|
||||
config['cvae']['encoder_depths'] = [3, 3, 3, 9, 3]
|
||||
config['cvae']['encoder_dims'] = [32, 64, 128, 256, 64]
|
||||
config['cvae']['decoder_depths'] = [3, 9, 3, 3, 3, 3]
|
||||
config['cvae']['decoder_dims'] = [3, 32, 128, 128, 256, 512]
|
||||
config['cvae']['decoder_weights_depths'] = [3, 9, 3, 3, 3]
|
||||
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()
|
||||
|
||||
Reference in New Issue
Block a user