How to adjust a Neural Networks metadata
import onnx onnx_model = onnx.load(onnxModelName)
Add a metadata key (without duplication):
def addMeta(onnx_model, key, value): for entry in onnx_model.metadata_props: if entry.key == key: entry.value = value return meta = onnx_model.metadata_props.add() meta.key = key meta.value = value
Save the onnx model to file:
with open(onnxFilePath, "wb") as f: f.write(onnx_model.SerializeToString())