search

onnx / onnx-tensorflow

Tensorflow Backend for ONNX
Other
1.27k stars 296 forks source link

Error in PredictCost() for the op: op: "Softmax" attr #1044

Open krishnanNuance opened 1 year ago

krishnanNuance commented 1 year ago

Describe the bug

The model converted from ONNX to TF on validation returns NaN as result.

To Reproduce

  1. export fairseq's transformer lm model to onnx
  2. export the onnx model to TF
  3. load the saved_model and validate with any src_token value
  4. the output array contains NaN

Python, ONNX, ONNX-TF, Tensorflow version