Implemented 1 cycle learning rate schedule

source/model_utils.py

@@ -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