dev(C4W3 ugl): rename class

Author: cfav-dev

Course 4 - Generative Deep Learning/W3/ungraded_labs/C4_W3_Lab_1_VAE_MNIST.ipynb

@@ -217,7 +217,7 @@
    },
    "outputs": [],
    "source": [
-    "class KLDLoss(tf.keras.layers.Layer):\n",
+    "class KLDLayer(tf.keras.layers.Layer):\n",
     "  def call(self, inputs):\n",
     "    \"\"\"Computes the KLD loss and adds it to the model\n",
     "\n",
@@ -336,7 +336,7 @@
     "  z = Sampling()((mu, sigma))\n",
     "\n",
     "  # feed mu and sigma to the KLD layer\n",
-    "  kl_loss = KLDLoss()((mu, sigma))\n",
+    "  kl_loss = KLDLayer()((mu, sigma))\n",
     "\n",
     "  # build the whole encoder model\n",
     "  model = tf.keras.Model(inputs=inputs, outputs=[z, kl_loss])\n",