...
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.
...