Update `test_workflow` to ONNX (#15974)

3 months ago · 18de47cda6
parent eeb733af32
commit 18de47cda6
1 changed files with 1 additions and 1 deletions
--- a/tests/test_python.py
+++ b/tests/test_python.py
@ -202,7 +202,7 @@ def test_workflow():
    model.train(data="coco8.yaml", epochs=1, imgsz=32, optimizer="SGD")
    model.val(imgsz=32)
    model.predict(SOURCE, imgsz=32)
-    model.export(format="torchscript")
+    model.export(format="onnx")


 def test_predict_callback_and_setup():