fixed feature sizes, cvae working

This commit is contained in:
2023-01-25 21:00:47 +00:00
parent cba9d8ce90
commit 46e37e22eb
12 changed files with 39 additions and 43 deletions
+3 -3
View File
@@ -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
+1 -1
View File
@@ -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']
+1 -1
View File
@@ -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']
+2 -2
View File
@@ -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']