You are viewing a plain text version of this content. The canonical link for it is here.
Posted to dev@mxnet.apache.org by Marek Kolodziej <mk...@gmail.com> on 2018/05/21 16:33:53 UTC
Proposal of MxNet-to-ONNX exporter
def mxnet.contrib.onnx._export.export_model( sym, params, input_shape,
output):
"""
Exports a given MXNet symbol object file or path to saved file to ONNX
model file.
Input Parameters -
----------
sym -
A str object (path to json file) or mxnet symbol object or
checkpointed mxnet symbol object
params -
A str object (path to params file) or dict object containing the
model params
input_shape -
list of tuple object , specifies the shape of each input to the
model
output -
path to the output file, including the filename. Default: current
path , filename: model.onnx
Return Type –
----------
onnx_model_path -
str object , path to saved .onnx file.
"""
...
return onnx_model_path
Re: Proposal of MxNet-to-ONNX exporter
Posted by Hagay Lupesko <lu...@gmail.com>.
Marek,
Can you add more details on your proposal?
(Implementation details, operator coverage, test coverage planned)
I am aware of similar work done by other contributors (Roshani, Rajan and
Anirudh) - how is this going to align? Should you guys work together?
Hagay
On Mon, May 21, 2018, 11:17 Marek Kolodziej <mk...@gmail.com> wrote:
> def mxnet.contrib.onnx._export.export_model( sym, params, input_shape,
> output):
> """
> Exports a given MXNet symbol object file or path to saved file to ONNX
> model file.
> Input Parameters -
> ----------
> sym -
> A str object (path to json file) or mxnet symbol object or
> checkpointed mxnet symbol object
> params -
> A str object (path to params file) or dict object containing the
> model params
> input_shape -
> list of tuple object , specifies the shape of each input to the
> model
> output -
> path to the output file, including the filename. Default: current
> path , filename: model.onnx
>
> Return Type –
> ----------
> onnx_model_path -
> str object , path to saved .onnx file.
> """
> ...
> return onnx_model_path
>