working auto encoders

2022-09-25 22:12:36 +00:00
parent 26916c00e7
commit 040b29e898
9 changed files with 41 additions and 71 deletions
@@ -64,7 +64,10 @@ class AEEncoder(tf.keras.Model):
        x = self.encoder(x, training=training)
        x = self.bottleneck(x, training=training)

-        return x
+        if len(inputs) == 2:
+            return x, inputs[1]
+        else:
+            return x


 class Decoder(tf.keras.Model):