Commit 672a2925 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 88b11438
......@@ -298,13 +298,7 @@ class SEQ_2_SEQ_GMVAE(HyperModel):
encoder = Model_E4(encoder)
encoder = BatchNormalization()(encoder)
z_cat = Dense(
self.number_of_components,
activation="softmax",
kernel_initializer=RandomNormal(
mean=(1 / self.number_of_components), stddev=0.05, seed=None
),
)(encoder)
z_cat = Dense(self.number_of_components, activation="softmax",)(encoder)
z_gauss = Dense(
tfpl.IndependentNormal.params_size(ENCODING * self.number_of_components),
activation=None,
......
......@@ -284,13 +284,7 @@ class SEQ_2_SEQ_GMVAE:
encoder = Model_E4(encoder)
encoder = BatchNormalization()(encoder)
z_cat = Dense(
self.number_of_components,
activation="softmax",
kernel_initializer=RandomNormal(
mean=(1 / self.number_of_components), stddev=0.05, seed=None
),
)(encoder)
z_cat = Dense(self.number_of_components, activation="softmax",)(encoder)
z_gauss = Dense(
tfpl.IndependentNormal.params_size(
self.ENCODING * self.number_of_components
......
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