Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Commit 5f32dcb

Browse files
authored
Updated yolo v3 tests (#57) (#58)
Updating yolo v3 tests before removing old yolo v3 model
1 parent bffcae6 commit 5f32dcb

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

tests/sparsezoo/models/detection/test_yolo.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,17 @@
2525
),
2626
[
2727
(True, "pytorch", "ultralytics", "coco", None, "base", "none", None),
28-
(True, "pytorch", "ultralytics", "coco", None, "pruned", "aggressive", None),
28+
(
29+
True,
30+
"pytorch",
31+
"ultralytics",
32+
"coco",
33+
None,
34+
"pruned_quant",
35+
"aggressive_94",
36+
None,
37+
),
38+
(True, "pytorch", "ultralytics", "coco", None, "pruned", "aggressive_97", None),
2939
],
3040
)
3141
def test_yolo_v3(

0 commit comments

Comments
 (0)