Update transformer_asr.ipynb

Author: divyashreepathihalli

examples/audio/ipynb/transformer_asr.ipynb

@@ -321,7 +321,7 @@
     "            preds = self([source, dec_input])\n",
     "            one_hot = tf.one_hot(dec_target, depth=self.num_classes)\n",
     "            mask = tf.math.logical_not(tf.math.equal(dec_target, 0))\n",
-    "            loss = model.compute_loss(None, one_hot, preds, sample_weight=mask)\n",
+    "            loss = self.compute_loss(None, one_hot, preds, sample_weight=mask)\n",
     "        trainable_vars = self.trainable_variables\n",
     "        gradients = tape.gradient(loss, trainable_vars)\n",
     "        self.optimizer.apply_gradients(zip(gradients, trainable_vars))\n",
@@ -336,7 +336,7 @@
     "        preds = self([source, dec_input])\n",
     "        one_hot = tf.one_hot(dec_target, depth=self.num_classes)\n",
     "        mask = tf.math.logical_not(tf.math.equal(dec_target, 0))\n",
-    "        loss = model.compute_loss(None, one_hot, preds, sample_weight=mask)\n",
+    "        loss = self.compute_loss(None, one_hot, preds, sample_weight=mask)\n",
     "        self.loss_metric.update_state(loss)\n",
     "        return {\"loss\": self.loss_metric.result()}\n",
     "\n",