save

2023-03-07 08:59:20 +00:00
parent 0b518e7f0e
commit e192f6f9e3
21 changed files with 1004 additions and 73 deletions
@@ -19,7 +19,7 @@ ablation_count = [config['use_skinning'], config['remove_global_orientation'], c
 assert ablation_count == 1 or ablation_count == 0

 config['gpu'] = '0'
-config['name'] = 'cvae_curv'
+config['name'] = 'cvae_curv_3'
 config['notes'] = ''
 config['tags'] = ['iccv', 'cvae_curv']

@@ -50,12 +50,13 @@ if config['is_cape']:

 config['latent_dim'] = 128

-config['pretrained_model'] = ''
+config['pretrained_model'] = 'checkpoints/iccv/cvae_curv'
+config['pretrained_model_iteration'] = ''

 ###############    
 config['lr'] = 0.0001#                 0.0001
 ###############
-config['starting_balancing_coeff'] = 0.01
+config['starting_balancing_coeff'] = 1.0

 config['signature_dynamic'] = { 'id': (), 'frame_nr': (),
                          'pose_shape': (72 + 11,),