[DLMED] update according to comments

Signed-off-by: Nic Ma <[email protected]>

Author: Nic-Ma

acceleration/fast_training_tutorial.ipynb

@@ -313,7 +313,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 45,
+   "execution_count": 6,
    "metadata": {
     "scrolled": true
    },
@@ -484,7 +484,7 @@
     "                    best_metrics_epochs_and_time[2].append(\n",
     "                        time.time() - total_start\n",
     "                    )\n",
-    "                    torch.save(model.state_dict(), \"best_metric_model.pt\")\n",
+    "                    torch.save(model.state_dict(), os.path.join(root_dir, \"best_metric_model.pt\"))\n",
     "                    print(\"saved new best metric model\")\n",
     "                print(\n",
     "                    f\"current epoch: {epoch + 1} current\"\n",