🧹 Correct output_names for coreml exports

yolo/utils/export_utils.py

@@ -36,8 +36,6 @@ def __init__(self, cfg: Config, model: YOLO, format: str, model_path: Optional[s
             "9_bbox_deltas_large",
         ]
 
-        self.output_names: List[str] = ["preds_cls", "preds_anc", "preds_box"]
-
     def export_onnx(self, dynamic_axes: Optional[Dict[str, Dict[int, str]]] = None, model_path: Optional[str] = None):
         logger.info(f":package: Exporting model to onnx format")
         import torch
@@ -93,6 +91,8 @@ def export_coreml(self):
         # Convert to Core ML program using the Unified Conversion API.
         logging.getLogger("coremltools").disabled = True
 
+        self.output_names: List[str] = ["preds_cls", "preds_anc", "preds_box"]
+
         model_from_export = ct.convert(
             exported_program, outputs=[ct.TensorType(name=name) for name in self.output_names], convert_to="mlprogram"
         )