Enable yolov5ts unittest for ONNX exporting

zhiqwang · Jan 19, 2022 · f1626c2 · f1626c2
1 parent 3485ea1
commit f1626c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_onnx.py b/test/test_onnx.py
@@ -110,7 +110,7 @@ def get_test_images(self):
         [
             ("yolov5s", "r3.1"),
             ("yolov5m", "r4.0"),
-            # ("yolov5ts", "r4.0"),
+            ("yolov5ts", "r4.0"),
         ],
     )
     def test_yolort_export_onnx(self, arch, upstream_version):