diff --git a/Train-Runpod.ipynb b/Train-Runpod.ipynb index 34175a0..9b9112c 100644 --- 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)" ] },