fix bugs for dialogue tutorial (#4211)

Signed-off-by: Zhilin Wang <[email protected]>
NVIDIA · Jun 3, 2022 · c9a5c1f · c9a5c1f
1 parent 5a33417
commit c9a5c1f
Showing 1 changed file with 26 additions and 26 deletions.
diff --git a/tutorials/nlp/Dialogue.ipynb b/tutorials/nlp/Dialogue.ipynb
@@ -28,7 +28,7 @@
       "source": [
         "import os \n",
         "!apt-get update && apt-get install -y libsndfile1 ffmpeg\n",
-        "!git clone https://github.com/NVIDIA/NeMo --branch main\n",
+        "!git clone https://github.com/NVIDIA/NeMo --branch r1.9.0\n",
         "os.chdir('NeMo')\n",
         "!./reinstall.sh\n",
         "os.chdir('..')\n"
@@ -87,7 +87,7 @@
         "\n",
         "An example is:\n",
         "\n",
-        "* utterance: what alarms have i set for tomorrow intent: \n",
+        "* utterance: what alarms have i set for tomorrow \n",
         "* intent: alarm_query\n",
         "* slots: date(tomorrow)\n",
         "\n",
@@ -287,15 +287,20 @@
     },
     {
       "cell_type": "markdown",
-      "source": [
-        "## 1.4 (Optional) To train/ test a GPT2 model on the assistant dataset, run the cell below "
-      ],
       "metadata": {
         "id": "-44x5PqyrOeQ"
-      }
+      },
+      "source": [
+        "## 1.4 (Optional) To train/ test a GPT2 model on the assistant dataset, run the cell below "
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "QyqQbpR4rNHT"
+      },
+      "outputs": [],
       "source": [
         "# model.dataset.data_dir: folder to load data from\n",
         "# model.dataset.dialogues_example_dir: folder that stores predictions for each sample\n",
@@ -312,15 +317,13 @@
         "  model.dataset.target_template=with_slots \\\n",
         "  model.dataset.eval_mode=generation \\\n",
         "  exp_manager.create_wandb_logger=False)"
-      ],
-      "metadata": {
-        "id": "QyqQbpR4rNHT"
-      },
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "FbQ-6TVM1yQg"
+      },
       "source": [
         "**After 1 epoch:**\n",
         "\n",
@@ -417,10 +420,7 @@
         "     test_loss_epoch       0.019178826361894608\n",
         "────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────\n",
         "```"
-      ],
-      "metadata": {
-        "id": "FbQ-6TVM1yQg"
-      }
+      ]
     },
     {
       "cell_type": "markdown",
@@ -437,7 +437,7 @@
         "\n",
         "An example is:\n",
         "\n",
-        "* utterance: I will be eating there at 11:30 am so make it for then.\n",
+        "* utterance: I will be eating there at 11:30 am so make the reservation for then.\n",
         "* intent: ReserveRestaurant\n",
         "* slots: {\"time\": \"11:30 am\"}\n",
         "\n",
@@ -498,14 +498,14 @@
     },
     {
       "cell_type": "code",
-      "source": [
-        "!ls sgd_gpt2_predictions"
-      ],
+      "execution_count": null,
       "metadata": {
         "id": "kGDlV5HvI2PQ"
       },
-      "execution_count": null,
-      "outputs": []
+      "outputs": [],
+      "source": [
+        "!ls sgd_gpt2_predictions"
+      ]
     },
     {
       "cell_type": "markdown",
@@ -515,7 +515,7 @@
       "source": [
         "**After 1 epoch:**\n",
         "\n",
-        "More epoches would needed to reach convergence.\n",
+        "More epochs would needed to reach convergence.\n",
         "\n",
         "\n",
         "```\n",
@@ -590,7 +590,7 @@
         "An example is \n",
         "\n",
         "\n",
-        "*   question: what county is nine mile in\n",
+        "*   question: What county is Nine Mile in?\n",
         "*   extracted_answer: Onondaga\n",
         "*   fluent_answer: Nine Mile is in Onondaga county.\n"
       ]
@@ -667,7 +667,7 @@
       "source": [
         "**After 1 epoch:**\n",
         "\n",
-        "Train more epoches for optimal performance\n",
+        "Train more epochs for optimal performance\n",
         "\n",
         "```\n",
         "────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────\n",
@@ -713,4 +713,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}