Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Addressed ONNX module documentation warnings and added notes for shor…
Browse files Browse the repository at this point in the history
…t-form representation (#13259)
  • Loading branch information
vdantu authored and sandeep-krishnamurthy committed Nov 15, 2018
1 parent ead3af2 commit 100a4aa
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 0 deletions.
5 changes: 5 additions & 0 deletions python/mxnet/contrib/onnx/mx2onnx/export_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,11 @@ def export_model(sym, params, input_shape, input_type=np.float32,
-------
onnx_file_path : str
Onnx file path
Notes
-----
This method is available when you ``import mxnet.contrib.onnx``
"""

try:
Expand Down
9 changes: 9 additions & 0 deletions python/mxnet/contrib/onnx/onnx2mx/import_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,11 @@ def import_model(model_file):
aux_params : dict of ``str`` to :class:`~mxnet.ndarray.NDArray`
Dict of converted parameters stored in ``mxnet.ndarray.NDArray`` format
Notes
-----
This method is available when you ``import mxnet.contrib.onnx``
"""
graph = GraphProto()

Expand All @@ -58,6 +63,10 @@ def get_model_metadata(model_file):
"""
Returns the name and shape information of input and output tensors of the given ONNX model file.
Notes
-----
This method is available when you ``import mxnet.contrib.onnx``
Parameters
----------
model_file : str
Expand Down
5 changes: 5 additions & 0 deletions python/mxnet/contrib/onnx/onnx2mx/import_to_gluon.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,11 @@ def import_to_gluon(model_file, ctx):
-------
sym_block : :class:`~mxnet.gluon.SymbolBlock`
A SymbolBlock object representing the given model file.
Notes
-----
This method is available when you ``import mxnet.contrib.onnx``
"""
graph = GraphProto()
try:
Expand Down

0 comments on commit 100a4aa

Please sign in to comment.