fixed feature sizes, cvae working

configs/config_ae_static.py

@@ -31,7 +31,7 @@ config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
 config['cond_ae']['activation'] = 'linear'
 
 
-config['gpu'] = '5'
+config['gpu'] = '0'
 config['lr'] = 0.0005
 config['name'] = 'static_post_4'
 config['notes'] = 'convnext + convnext, | \

configs/review/config_abl_all.py

@@ -43,9 +43,9 @@ config['test_dir'] = '/home/workspace/data/dataset/test_ready'
 # config['frames_file_valid'] = 'frames_dress_30k.txt'
 # config['frames_file_test'] = 'frames_dress_30k.txt'
 
-config['static_model'] = 'checkpoints/post/static_r7'
-config['body_model'] = 'checkpoints/post/body_r7'
-config['curvature_model'] = 'checkpoints/post/curv_r7'
+config['static_model'] = 'checkpoints/post/static_r7_2'
+config['body_model'] = 'checkpoints/post/body_r7_2'
+config['curvature_model'] = 'checkpoints/post/curv_r7_2'
 
 
 config['smpl_res'] = 14475

configs/review/config_ae_body.py

@@ -33,7 +33,7 @@ config['cond_ae']['activation'] = 'tanh'
 
 config['gpu'] = '4'
 config['lr'] = 0.0001
-config['name'] = 'body_r7'
+config['name'] = 'body_r7_2'
 config['notes'] = 'resnet18' 
 config['tags'] = ['review_7']
 

configs/review/config_ae_curv.py

@@ -33,7 +33,7 @@ config['cond_ae']['activation'] = 'tanh'
 
 config['gpu'] = '3'
 config['lr'] = 0.001
-config['name'] = 'curv_r7'
+config['name'] = 'curv_r7_2'
 config['notes'] = 'resnet18' 
 config['tags'] = ['review_7']
 

configs/review/config_ae_static.py

@@ -31,9 +31,9 @@ config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
 config['cond_ae']['activation'] = 'linear'
 
 
-config['gpu'] = '6'
+config['gpu'] = '0'
 config['lr'] = 0.0005
-config['name'] = 'static_r7'
+config['name'] = 'static_r7_2'
 config['notes'] = 'resnet18 based'
 config['tags'] = ['review_7']