From 32aecaea363d88fde4bd06400a15844362074754 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 12 Apr 2019 14:21:47 +0200 Subject: [PATCH] Deactivated pooling and upsampling layers Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/model.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/twomartens/masterthesis/aae/model.py b/src/twomartens/masterthesis/aae/model.py index 5ef34ce..0263e7f 100644 --- a/src/twomartens/masterthesis/aae/model.py +++ b/src/twomartens/masterthesis/aae/model.py @@ -65,13 +65,13 @@ class Encoder(keras.Model): result = self.conv1(inputs) result = self.conv1_a(result) # result = self.dropout(result) - result = self.pool1(result) + # result = self.pool1(result) result = self.conv3(result) result = self.conv3_a(result) - result = self.pool3(result) + # result = self.pool3(result) result = self.conv4(result) result = self.conv4_a(result) - result = self.pool4(result) + # result = self.pool4(result) return result @@ -107,13 +107,13 @@ class Decoder(keras.Model): """See base class.""" result = self.deconv1(inputs) result = self.deconv1_a(result) - result = self.upsample1(result) + # result = self.upsample1(result) result = self.deconv2(result) result = self.deconv2_a(result) - result = self.upsample2(result) + # result = self.upsample2(result) result = self.deconv3(result) result = self.deconv3_a(result) - result = self.upsample3(result) + # result = self.upsample3(result) result = self.deconv5(result) result = k.sigmoid(result)