Implemented 1 cycle learning rate schedule

source/model_utils.py

@@ -336,7 +336,7 @@ class Entropy_regulariser(Layer):
     def call(self, z, **kwargs):
 
         entropy = K.sum(
-            tf.multiply(z, tf.where(~tf.math.is_inf(K.log(z)), K.log(z), 0)), axis=0
+            tf.multiply(z, tf.where(~tf.math.is_inf(K.log(z)), K.log(z), 0)), axis=1
         )
 
         # Adds metric that monitors dead neurons in the latent space