add pretrained option for

2022-10-29 20:05:13 +00:00
parent ef05cf57d4
commit 6dd16c7002
2 changed files with 7 additions and 3 deletions
@@ -11,12 +11,12 @@ config['notes'] = 'convnext + convnext, add dense with tanh at the end of revers
    ' 

 config['cond_ae']['upscale_output'] = False
-config['batch_size'] = 32
+config['pretrained_model'] = ''


 config['signature_dynamic'] = { 'id': (), }
 config['signature_static'] = { 'uv_static': (512, 256, 3),
-                         'mesh_static': (14475, 3,),
-                         'mesh_mask': (14475,),
+                         'mesh_static': (config['smpl_res'], 3,),
+                         'mesh_mask': (config['smpl_res'],),
                         'uv_mask': (512, 256,),
                        }