Commit 70b60e20 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent ef969160
...@@ -345,7 +345,7 @@ class SEQ_2_SEQ_GMVAE: ...@@ -345,7 +345,7 @@ class SEQ_2_SEQ_GMVAE:
z = MMDiscrepancyLayer(prior=self.prior, beta=mmd_beta)(z) z = MMDiscrepancyLayer(prior=self.prior, beta=mmd_beta)(z)
# Identity layer controlling clustering and latent space statistics # Identity layer controlling clustering and latent space statistics
#z = Latent_space_control()(z, z_gauss, z_cat) z = Latent_space_control()(z, z_gauss, z_cat)
# Define and instantiate generator # Define and instantiate generator
generator = Model_D1(z) generator = Model_D1(z)
......
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