diff --git a/mediapipe/model_maker/python/vision/image_classifier/image_classifier.py b/mediapipe/model_maker/python/vision/image_classifier/image_classifier.py index 1ff6132b4..df71a8fef 100644 --- a/mediapipe/model_maker/python/vision/image_classifier/image_classifier.py +++ b/mediapipe/model_maker/python/vision/image_classifier/image_classifier.py @@ -177,7 +177,7 @@ class ImageClassifier(classifier.Classifier): Args: 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. """ if not tf.io.gfile.exists(self._hparams.export_dir):