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

Commit

Permalink
fix sanity
Browse files Browse the repository at this point in the history
  • Loading branch information
Wei Chu committed Jan 27, 2021
1 parent 2c5e386 commit f4b0c37
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/mxnet/contrib/onnx/mx2onnx/export_onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,8 +154,8 @@ def get_outputs(sym, params, in_shape, in_label, in_type):
## Infer output types
# Remove any input listed in params from sym.list_inputs() and bind them to the input types provided
# by user. Also remove in_label
in_dtype = {n: mapping.TENSOR_TYPE_TO_NP_TYPE[t] for n, t in zip([n for n in sym.list_inputs()
if n not in params and n != in_label], in_type)}
in_dtype = {n: mapping.TENSOR_TYPE_TO_NP_TYPE[t]
for n, t in zip([n for n in sym.list_inputs() if n not in params and n != in_label], in_type)}
# Add params and their shape to list of inputs
in_dtype.update({n: v.dtype for n, v in params.items() if n in sym.list_inputs()})
_, out_type, _ = sym.infer_type(**in_dtype)
Expand Down

0 comments on commit f4b0c37

Please sign in to comment.