Including additional metadata when using torch.onnx.export

[Mayur28]

Hi,

I am trying to export a model from PyTorch to ONNX, however, I am trying to do so while also trying to additional metadata in the model. Specifically, in addition to converting the model to ONNX, I just wanted to find out if there is a way to perhaps include a dictionary of metadata when the model is exported so that when the ONNX model is loaded, I am able to also access the dictionary of metadata?

Thanks!