Update Use_WandbModelCheckpoint_in_your_Keras_workflow.ipynb (#611)

knisar · web-flow · commit 8278022e6a92 · 2025-09-23T14:33:14.000-07:00
diff --git a/colabs/keras/Use_WandbModelCheckpoint_in_your_Keras_workflow.ipynb b/colabs/keras/Use_WandbModelCheckpoint_in_your_Keras_workflow.ipynb
@@ -281,7 +281,7 @@
     "    validation_data = validloader,\n",
     "    callbacks = [\n",
     "        WandbMetricsLogger(log_freq=10),\n",
-    "        WandbModelCheckpoint(filepath=\"models/\") # Notice the use of WandbModelCheckpoint here\n",
+    "        WandbModelCheckpoint(filepath=\"models/model.keras\") # Notice the use of WandbModelCheckpoint here\n",
     "    ]\n",
     ")\n",
     "\n",
