UltralyticsAssistant 1 month ago
parent 9bd4de7431
commit efdd847fe1
  1. 1
      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()

Loading…
Cancel
Save