[yolov8] standardize to export models as 'model.onnx' (#1867) (#1868)

neuralmagic · Nov 29, 2023 · c052e27 · c052e27
1 parent 8b5d66e
commit c052e27
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/sparseml/yolov8/trainers.py b/src/sparseml/yolov8/trainers.py
@@ -727,12 +727,8 @@ def export(self, **kwargs):
                 else manager
             )
 
-        if args.get("name") and args["name"]:
-            name = args["name"]
-        else:
-            name = f"{type(self.model).__name__}.onnx"
-
         save_dir = args["save_dir"]
+        name = "model.onnx"  # save 'model.onnx' in deployment directory
         if not os.path.exists(save_dir):
             os.mkdir(save_dir)