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

Update the docstring to make it consistent with the model option update.

Browse Source 
PiperOrigin-RevId: 488761331
This commit is contained in:
MediaPipe Team 2022-11-15 14:34:45 -08:00 committed by Copybara-Service
parent 496720308c
commit e65f21e2d8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/model_maker/python/vision/image_classifier/image_classifier.py
View File

@ -177,7 +177,7 @@ class ImageClassifier(classifier.Classifier):
Args: Args:
model_name: File name to save TFLite model with metadata. The full export model_name: File name to save TFLite model with metadata. The full export
path is {self._hparams.model_dir}/{model_name}. path is {self._hparams.export_dir}/{model_name}.
quantization_config: The configuration for model quantization. quantization_config: The configuration for model quantization.
""" """
if not tf.io.gfile.exists(self._hparams.export_dir): if not tf.io.gfile.exists(self._hparams.export_dir):