Skip to content

GitLab

Commit 092a5461 authored by Raymond Yuan's avatar Raymond Yuan
Browse files

more minor updates

parent 41b8e351
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
samples/outreach/blogs/segmentation_blogpost/image_segmentation.ipynb
View file @ 092a5461
...@@ -1068,7 +1068,7 @@ ...@@ -1068,7 +1068,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"save_model_path = '/tmp/weights.hdf5'\n", "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 @@ ...@@ -1119,7 +1119,7 @@
}, },
"outputs": [], "outputs": [],
"source": [ "source": [
"dice = = history.history['dice_loss']\n", "dice = history.history['dice_loss']\n",
"val_dice = history.history['val_dice_loss']\n", "val_dice = history.history['val_dice_loss']\n",
"\n", "\n",
"loss = history.history['loss']\n", "loss = history.history['loss']\n",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment