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

Implemented weight saving callback in

parent ef969160
......@@ -345,7 +345,7 @@ class SEQ_2_SEQ_GMVAE:
z = MMDiscrepancyLayer(prior=self.prior, beta=mmd_beta)(z)
# 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
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