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

Object Detector add batch_size and train_data to get_steps_per_epoch.

Browse Source 
PiperOrigin-RevId: 522149938
This commit is contained in:
MediaPipe Team 2023-04-05 14:09:08 -07:00 committed by Copybara-Service
parent 5615c1e459
commit 6605f551e7
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/model_maker/python/vision/object_detector/object_detector.py
View File

@ -105,7 +105,9 @@ class ObjectDetector(classifier.Classifier):
""" """
self._optimizer = self._create_optimizer( self._optimizer = self._create_optimizer(
model_util.get_steps_per_epoch( model_util.get_steps_per_epoch(
self._hparams.steps_per_epoch, steps_per_epoch=self._hparams.steps_per_epoch,
batch_size=self._hparams.batch_size,
train_data=train_data,
) )
) )
self._create_model() self._create_model()