update

2022-09-13 09:23:01 +00:00
parent 977e49900d
commit 440444cd82
19 changed files with 1630 additions and 81 deletions
@@ -43,7 +43,7 @@ from auto_encoder_body import AutoEncoder as BodyAE
 from auto_encoder_curv import AutoEncoder as CurvatureAE
 from auto_encoder_static import AutoEncoder as StaticAE
 from cvae_curv import CVAE as CVAECurv
-from models.cvae import CVAE as CVAEMain
+from models.cvae_model import CVAE as CVAEMain

 name='cvae_12_6'
 def init_wandb():