From a28c9d2c2697a387795d0cd721c99024923dbd18 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Tue, 25 Oct 2022 13:41:41 -0700 Subject: [PATCH] Set `steps_per_epoch` to None when calling model.fit() method for image classifier. PiperOrigin-RevId: 483764377 --- .../python/vision/image_classifier/train_image_classifier_lib.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mediapipe/model_maker/python/vision/image_classifier/train_image_classifier_lib.py b/mediapipe/model_maker/python/vision/image_classifier/train_image_classifier_lib.py index 704d71a5a..265c36a6e 100644 --- a/mediapipe/model_maker/python/vision/image_classifier/train_image_classifier_lib.py +++ b/mediapipe/model_maker/python/vision/image_classifier/train_image_classifier_lib.py @@ -98,6 +98,5 @@ def train_model(model: tf.keras.Model, hparams: hp.HParams, return model.fit( x=train_ds, epochs=hparams.train_epochs, - steps_per_epoch=hparams.steps_per_epoch, validation_data=validation_ds, callbacks=callbacks)