Rename to test_yolort_export_onnx

zhiqwang · Aug 25, 2021 · 2a1d997 · 2a1d997
1 parent a123bc5
commit 2a1d997
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_onnx.py b/test/test_onnx.py
@@ -105,7 +105,7 @@ def get_test_images(self) -> Tuple[List[Tensor], List[Tensor]]:
         ('yolov5m', 'r4.0'),
         ('yolotr', 'r4.0'),
     ])
-    def test_yolov5s_r31(self, arch, upstream_version):
+    def test_yolort_export_onnx(self, arch, upstream_version):
         images_one, images_two = self.get_test_images()
         images_dummy = [torch.ones(3, 100, 100) * 0.3]