Commit 4516d533 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added a hyperparameter controlling the number of layers near the latent space

parent 41670663
Pipeline #87831 passed with stage
in 16 minutes and 13 seconds
......@@ -381,7 +381,7 @@ class SEQ_2_SEQ_GMVAE:
kernel_initializer=he_uniform(),
use_bias=True,
)
for _ in self.dense_layers_per_branch
for _ in range(self.dense_layers_per_branch)
]
# Decoder layers
......@@ -396,7 +396,7 @@ class SEQ_2_SEQ_GMVAE:
kernel_initializer=he_uniform(),
use_bias=True,
)
for _ in self.dense_layers_per_branch
for _ in range(self.dense_layers_per_branch)
]
Model_D2 = Dense(
self.DENSE_1,
......
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