more minor updates

092a5461 · Raymond Yuan · 41b8e351 · 092a5461
Commit 092a5461 authored Aug 02, 2018 by Raymond Yuan
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

samples/outreach/blogs/segmentation_blogpost/image_segmentation.ipynb ...each/blogs/segmentation_blogpost/image_segmentation.ipynb +2 -2

No files found.
--- a/samples/outreach/blogs/segmentation_blogpost/image_segmentation.ipynb
+++ b/samples/outreach/blogs/segmentation_blogpost/image_segmentation.ipynb
@@ -1068,7 +1068,7 @@
   "outputs": [],
   "source": [
    "save_model_path = '/tmp/weights.hdf5'\n",
-    "cp = tf.keras.callbacks.ModelCheckpoint(filepath=save_model_path, monitor='val_dice_loss', mode='max', save_best_only=True)"
+    "cp = tf.keras.callbacks.ModelCheckpoint(filepath=save_model_path, monitor='val_dice_loss', save_best_only=True, verbose=1)"
   ]
  },
  {
@@ -1119,7 +1119,7 @@
   },
   "outputs": [],
   "source": [
-    "dice = = history.history['dice_loss']\n",
+    "dice = history.history['dice_loss']\n",
    "val_dice = history.history['val_dice_loss']\n",
    "\n",
    "loss = history.history['loss']\n",