...
| Code Block | ||||||
|---|---|---|---|---|---|---|
| ||||||
def mxnet.contrib.onnx.get_model_metadata(model_file):
"""
Returns the metadata information of the given model. Currently it will return the shape information of the input and output tensors.
Input Parameters –
----------
model_file –
A string object representing the path to the ONNX model file.
Return Type –
----------
model_metadata –
A dictionary object mapping various metadata to its corresponding value. The proposed implementation will return a dictionary with following template.
{
“input_tensor_shape”: <list of tuples representing the shape of the input paramters>,
“output_tensor_shape”: <list of tuples representing the shape of the output of the model>
}
"""
...
return metadata |
...
Export MXNet symbolic model to an ONNX model file.
Given a MXNet model object/file, export the model to ONNX model file.
...