fix the train cell in notebook

2022-11-26 13:30:23 -05:00 · 2022-11-26 13:30:23 -05:00 · d0ae3b1ce9
parent 794e034988
commit d0ae3b1ce9
1 changed files with 1 additions and 1 deletions
--- a/Train-Runpod.ipynb
+++ b/Train-Runpod.ipynb
@ -238,7 +238,7 @@
   "outputs": [],
   "source": [
    "# run the trainer, wait until it finishes then SCROLL DOWN to the next cell\n",
-    "!python main.py --base {my_yaml} -t  --actual_resume whatever.ckpt -n test --data_root input\n",
+    "!python main.py --base {my_yaml} -t  --actual_resume {downloaded_model_path} -n test --data_root input\n",
    "print(finish_msg)"
   ]
  },