Commit 4a0a0791 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added Batch Normalization to SEQ2SEQ_VAE

parent b8e54032
This diff is collapsed.
......@@ -223,11 +223,16 @@ class SEQ_2_SEQ_VAE:
# Define and instanciate encoder
x = Input(shape=self.input_shape[1:])
encoder = Model_E0(x)
encoder = BatchNormalization()(encoder)
encoder = Model_E1(encoder)
encoder = BatchNormalization()(encoder)
encoder = Model_E2(encoder)
encoder = BatchNormalization()(encoder)
encoder = Model_E3(encoder)
encoder = BatchNormalization()(encoder)
encoder = Dropout(self.DROPOUT_RATE)(encoder)
encoder = Model_E4(encoder)
encoder = BatchNormalization()(encoder)
encoder = Model_E5(encoder)
z_mean = Dense(self.ENCODING)(encoder)
......@@ -243,10 +248,15 @@ class SEQ_2_SEQ_VAE:
# Define and instanciate generator
generator = Model_D0(z)
generator = BatchNormalization()(generator)
generator = Model_D1(generator)
generator = BatchNormalization()(generator)
generator = Model_D2(generator)
generator = BatchNormalization()(generator)
generator = Model_D3(generator)
generator = BatchNormalization()(generator)
generator = Model_D4(generator)
generator = BatchNormalization()(generator)
generator = Model_D5(generator)
x_decoded_mean = TimeDistributed(Dense(self.input_shape[2]))(generator)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment