diff --git a/WineQuality-lightning/03-DNN-Wine-Regression-lightning.ipynb b/WineQuality-lightning/03-DNN-Wine-Regression-lightning.ipynb index 7f34013735600894fbd148685a1fac5ee02eb453..8f3142145a116600e4e1f629faa3ac9002fa38e0 100644 --- a/WineQuality-lightning/03-DNN-Wine-Regression-lightning.ipynb +++ b/WineQuality-lightning/03-DNN-Wine-Regression-lightning.ipynb @@ -455,7 +455,7 @@ "metadata": {}, "outputs": [], "source": [ - "loaded_model = model.load_from_checkpoint('./run/models/best_model.ckpt/epoch=97-step=6272.ckpt')\n", + "loaded_model = model.load_from_checkpoint('./run/models/best_model.ckpt')\n", "print(\"Loaded:\")\n", "print(loaded_model)\n" ]