Commit 26a21aa0 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented 1 cycle learning rate schedule

parent f7468519
......@@ -335,6 +335,8 @@ class Entropy_regulariser(Layer):
def call(self, z, **kwargs):
# axis=1 increases the entropy of a cluster across instances
# axis=0 increases the entropy of the assignment for a given instance
entropy = K.sum(tf.multiply(z + 1e-5, tf.math.log(z) + 1e-5), axis=1)
# Adds metric that monitors dead neurons in the latent space
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