From a5bcb97d888df5ee721b80c37dea601291d888c0 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Tue, 8 Nov 2022 18:27:53 -0800 Subject: [PATCH] Adds an `AverageWordVecModel`. PiperOrigin-RevId: 487104909 --- mediapipe/model_maker/python/core/utils/model_util.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mediapipe/model_maker/python/core/utils/model_util.py b/mediapipe/model_maker/python/core/utils/model_util.py index 01e301e43..f10d9390c 100644 --- a/mediapipe/model_maker/python/core/utils/model_util.py +++ b/mediapipe/model_maker/python/core/utils/model_util.py @@ -39,11 +39,10 @@ def get_default_callbacks( """Gets default callbacks.""" summary_dir = os.path.join(export_dir, 'summaries') summary_callback = tf.keras.callbacks.TensorBoard(summary_dir) - # Save checkpoint every 20 epochs. checkpoint_path = os.path.join(export_dir, 'checkpoint') checkpoint_callback = tf.keras.callbacks.ModelCheckpoint( - checkpoint_path, save_weights_only=True, period=20) + checkpoint_path, save_weights_only=True) return [summary_callback, checkpoint_callback]