more minor updates

YongKun-Liu · Aug 2, 2018 · 092a546 · 092a546
1 parent 41b8e35
commit 092a546
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",