large refactor

2022-09-25 08:31:13 +00:00
parent 84ec5a056e
commit 92891f5b26
33 changed files with 1661 additions and 1635 deletions
@@ -80,12 +80,12 @@ class CVAE(tf.keras.Model):
        super(CVAE, self).__init__()
        self.latent_dim = latent_dim

-        model : StaticAE = StaticAE()
+        model : StaticAE = StaticAE(size=512)
        load_model(model, static_model)
        self.static_enc = tf.keras.Sequential([model.encoder, model.bottleneck])
        self.static_enc.trainable = False

-        model : BodyAE = BodyAE()
+        model : BodyAE = BodyAE(size=512)
        load_model(model, body_model)
        self.body_enc = tf.keras.Sequential([model.encoder, model.bottleneck])
        self.body_enc.trainable = False