Skip to content

Commit

Permalink
update yolo test
Browse files Browse the repository at this point in the history
  • Loading branch information
inisis committed Sep 8, 2024
1 parent c66ee66 commit 5c9265a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ def test_yolov10_export_onnx_matrix(self, request, task, dynamic, int8, half, ba
"task, dynamic, int8, half, batch",
product(["yolov10n"], [False], [False], [False], [1, 2]),
)
def test_yolov10_export_onnx_matrix(self, request, task, dynamic, int8, half, batch):
def test_yolov10_export_tflite_matrix(self, request, task, dynamic, int8, half, batch):
"""Exports YOLOv10 models to TFLite and tests inference with varying configurations."""
file = YOLO(task).export(
format="tflite",
Expand Down

0 comments on commit 5c9265a

Please sign in to comment.