cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal change for Model Maker

Browse Source 
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
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/model_maker/python/core/tasks/classifier.py
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)