Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

mx.gluon.import(input_file, in_format=‘onnx’)

mx.gluon.export(format=’onnx’, filename_prefix=”model_name”)

If any pre-processing/post-processing logic is required for gluon specific models, it can go under this gluon wrapper. These functions will internally call `serde` module APIs.

e.g. `mx.gluon.import(input_file, in_format=‘onnx’)` will internally call `mx.serde.import(input_file, in_format=‘onnx’, out_format='gluon')`

 

  •  For Symbolic interface

sym, params = mx.mod.Module.import(input_file, in_format=‘onnx’)

...