Commit 806428a3 authored by lucas_miranda's avatar lucas_miranda
Browse files

Early stopping now uses the overall loss function instead of the intercomponent overlap

parent afb3a322
......@@ -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(self.weight * K.sum(entropy), inputs=[z])
self.add_loss(self.weight * K.sum(entropy), inputs=[z])
return z
......@@ -173,7 +173,7 @@ class SEQ_2_SEQ_GMVAE:
number_of_components=1,
predictor=True,
overlap_loss=False,
entropy_reg_weight=1,
entropy_reg_weight=0.,
):
self.input_shape = input_shape
self.batch_size = batch_size
......
Markdown is supported
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