Fix documentation for checkpoint arg

PiperOrigin-RevId: 330016073

research/object_detection/export_tflite_graph_tf2.py

@@ -41,7 +41,7 @@ Example Usage:
 --------------
 python object_detection/export_tflite_graph_tf2.py \
     --pipeline_config_path path/to/ssd_model/pipeline.config \
-    --trained_checkpoint_prefix path/to/ssd_model/checkpoint \
+    --trained_checkpoint_dir path/to/ssd_model/checkpoint \
     --output_directory path/to/exported_model_directory
 
 The expected output SavedModel would be in the directory
@@ -57,7 +57,7 @@ Example Usage (in which we change the NMS iou_threshold to be 0.5 and
 NMS score_threshold to be 0.0):
 python object_detection/export_tflite_model_tf2.py \
     --pipeline_config_path path/to/ssd_model/pipeline.config \
-    --trained_checkpoint_prefix path/to/ssd_model/checkpoint \
+    --trained_checkpoint_dir path/to/ssd_model/checkpoint \
     --output_directory path/to/exported_model_directory
     --config_override " \
             model{ \

research/object_detection/g3doc/running_on_mobile_tf2.md

@@ -50,7 +50,7 @@ To use the script:
 # From the tensorflow/models/research/ directory
 python object_detection/export_tflite_graph_tf2.py \
     --pipeline_config_path path/to/ssd_model/pipeline.config \
-    --trained_checkpoint_prefix path/to/ssd_model/checkpoint \
+    --trained_checkpoint_dir path/to/ssd_model/checkpoint \
     --output_directory path/to/exported_model_directory
 ```