train encoder on raw curvature values

This commit is contained in:
2023-01-27 11:13:10 +00:00
parent a3c5b526b8
commit 6e6b0a6604
8 changed files with 60 additions and 71 deletions
+5 -9
View File
@@ -31,22 +31,18 @@ config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
config['cond_ae']['activation'] = 'tanh'
config['gpu'] = '3'
config['gpu'] = '1'
config['lr'] = 0.001
config['name'] = 'curv_post_1'
config['notes'] = 'convnext + convnext, | \
\
| \
deeper model | \
| \
'
config['name'] = 'curv_raw_r2'
config['notes'] = ''
config['tags'] = ['review_2']
config['cond_ae']['upscale_output'] = False
config['pretrained_model'] = ''
config['signature_dynamic'] = { 'id': (), 'frame_nr': (),
'uv_curvature': (512, 256, 7),
'uv_curvature': (512, 256, 1),
'mesh_curvature': (config['smpl_res'])
}
config['signature_static'] = { 'mesh_mask': (config['smpl_res'],),