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

Added maximum entropy regulariser to the gaussian mixture weight layer

parent 69146f0b
......@@ -292,6 +292,6 @@ class Entropy_regulariser(Layer):
# Adds metric that monitors dead neurons in the latent space
self.add_metric(-entropy, aggregation="mean", name="weight_entropy")
self.add_loss(-K.mean(entropy), inputs=[z])
self.add_loss(K.sum(entropy), inputs=[z])
return 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