Internal change for Model Maker

PiperOrigin-RevId: 489540387
This commit is contained in:
MediaPipe Team 2022-11-18 12:45:56 -08:00 committed by Copybara-Service
parent 1b594a0310
commit 524ac3ca61

View File

@ -91,6 +91,10 @@ class Classifier(custom_model.CustomModel):
self._history = self._model.fit( self._history = self._model.fit(
x=train_dataset, x=train_dataset,
epochs=self._hparams.epochs, epochs=self._hparams.epochs,
# `steps_per_epoch` is intentionally set to None in case the dataset
# is not repeated. Otherwise, the training process will stop when the
# dataset is exhausted even if there are epochs remaining.
steps_per_epoch=None,
validation_data=validation_dataset, validation_data=validation_dataset,
callbacks=self._callbacks) callbacks=self._callbacks)