Set steps_per_epoch to None when calling model.fit() method for image classifier.

PiperOrigin-RevId: 483764377
2022-10-25 13:41:41 -07:00 · 2022-10-25 13:41:41 -07:00 · a28c9d2c26
commit a28c9d2c26
parent 36bd9abb8f
1 changed files with 0 additions and 1 deletions
--- 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)