diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 32dea5983..8a90817e6 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -1164,6 +1164,7 @@ class Exporter: mct.exporter.pytorch_export_model(model=quant_model, save_model_path=f, repr_dataset=representative_dataset_gen) import onnx + model_onnx = onnx.load(f) # load onnx model for k, v in self.metadata.items(): meta = model_onnx.metadata_props.add()