Commit 9324194c authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 4a99d9c8
......@@ -191,7 +191,12 @@ class SEQ_2_SEQ_GMVAE:
),
[
tfd.Independent(
tfd.Normal(loc=tf.random.normal(self.ENCODING), scale=1),
tfd.Normal(
loc=tf.random.normal(
shape=[self.ENCODING], stddev=1 / self.ENCODING
),
scale=1,
),
reinterpreted_batch_ndims=1,
)
for _ in range(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