Fixed bug in trainer notebook (microsoft#434)

yichiac · Feb 27, 2022 · 579f578 · 579f578
1 parent 230e508
commit 579f578
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/tutorials/trainers.ipynb b/docs/tutorials/trainers.ipynb
@@ -492,13 +492,13 @@
     "        csv_reader = csv.DictReader(f, delimiter=',')\n",
     "        for i, row in enumerate(csv_reader):\n",
     "            try:\n",
-    "                train_rmse.append(float(row[\"train_rmse\"]))\n",
+    "                train_rmse.append(float(row[\"train_RMSE\"]))\n",
     "                train_steps.append(i)\n",
     "            except ValueError: # Ignore rows where train RMSE is empty\n",
     "                pass\n",
     "\n",
     "            try:\n",
-    "                val_rmse.append(float(row[\"val_rmse\"]))\n",
+    "                val_rmse.append(float(row[\"val_RMSE\"]))\n",
     "                val_steps.append(i)\n",
     "            except ValueError: # Ignore rows where val RMSE is empty\n",
     "                pass"