From 118d3d8f4754519b62567aab62bd9635fd54e4c3 Mon Sep 17 00:00:00 2001
From: Tek Kshetri <39838116+iamtekson@users.noreply.github.com>
Date: Tue, 16 Nov 2021 15:46:14 +0700
Subject: [PATCH] checkpointer added to the callback,
 076-077-078-Unet_nuclie_tutorial

---
 076-077-078-Unet_nuclei_tutorial.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/076-077-078-Unet_nuclei_tutorial.py b/076-077-078-Unet_nuclei_tutorial.py
index 5ce0ec3d0..4d97005e4 100644
--- a/076-077-078-Unet_nuclei_tutorial.py
+++ b/076-077-078-Unet_nuclei_tutorial.py
@@ -142,7 +142,8 @@
 
 callbacks = [
         tf.keras.callbacks.EarlyStopping(patience=2, monitor='val_loss'),
-        tf.keras.callbacks.TensorBoard(log_dir='logs')]
+        tf.keras.callbacks.TensorBoard(log_dir='logs'),
+        checkpointer]
 
 results = model.fit(X_train, Y_train, validation_split=0.1, batch_size=16, epochs=25, callbacks=callbacks)