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

PiperOrigin-RevId: 483764377
This commit is contained in:
MediaPipe Team 2022-10-25 13:41:41 -07:00 committed by Copybara-Service
parent 36bd9abb8f
commit a28c9d2c26

View File

@ -98,6 +98,5 @@ def train_model(model: tf.keras.Model, hparams: hp.HParams,
return model.fit( return model.fit(
x=train_ds, x=train_ds,
epochs=hparams.train_epochs, epochs=hparams.train_epochs,
steps_per_epoch=hparams.steps_per_epoch,
validation_data=validation_ds, validation_data=validation_ds,
callbacks=callbacks) callbacks=callbacks)