commit all

configs/config_cvae_curv.py

@@ -1,8 +1,12 @@
 config = dict()
 
 config['gpu'] = '2'
-config['name'] = 'cvae_curv'
-config['notes'] = ''
+config['name'] = 'cvae_curv_8'
+config['notes'] = 'no activation for decoder, softmax at the very end, lower balancing, output only 6 values, concatenate background | \
+  same as 3 but use ce loss | \
+  add ce mesh loss | \
+  higher balancing coeff | \
+  start from curv_5, but higher balancing  '
 
 config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
 
@@ -27,12 +31,13 @@ if config['is_cape']:
 
 config['latent_dim'] = 128
 
-config['pretrained_model'] = ''
+config['pretrained_model'] = 'checkpoints/cvae_curv_5'
+config['pretrained_model_iteration'] = '100000'
 
 ###############    
 config['lr'] = 0.0001#                 0.0001
 ###############
-config['starting_balancing_coeff'] = 0.1
+config['starting_balancing_coeff'] = 0.25
 
 config['signature_dynamic'] = { 'id': (), 'frame_nr': (),
                           'pose_shape': (72 + 11,),