Update Train_Colab.ipynb

fix bad variable in colab notebook
2023-06-04 00:10:45 -04:00 · 2023-06-04 00:10:45 -04:00 · 9970ccc3fb
parent 1155a28867
commit 9970ccc3fb
1 changed files with 2 additions and 2 deletions
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@ -149,7 +149,7 @@
        "    !wget $MODEL_LOCATION\n",
        "    clear_output()\n",
        "    print(\"DONE!\")\n",
-        "    download_path = os.path.join(os.getcwd(), os.path.basename(MODEL_URL))\n",
+        "    download_path = os.path.join(os.getcwd(), os.path.basename(MODEL_LOCATION))\n",
        "\n",
        "else:\n",
        "  save_name = MODEL_LOCATION\n",