mausmaux

Try this: ``` class LatentLossRegularizer(keras.layers.Layer): def call(self, X): codings_mean, codings_log_var = X latent_loss = -0.5 * tf.reduce_sum(1 + codings_log_var - tf.exp(codings_log_var) - tf.square(codings_mean), axis=-1) self.add_loss(tf.reduce_mean(latent_loss) / 784.) return X ```...