Commit ef969160 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent 3bc7e991
...@@ -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