Fixed bug in trainer notebook (#434)

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"