Skip to content

Instantly share code, notes, and snippets.

@jnovikov
Created August 17, 2022 20:08
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jnovikov/02a9aff9bf2188033e77bd91ff062856 to your computer and use it in GitHub Desktop.
Save jnovikov/02a9aff9bf2188033e77bd91ff062856 to your computer and use it in GitHub Desktop.
Onnx runtime poc
Pls download the https://drive.google.com/file/d/17ZzAfb_VLAAd_Xto9_Mb98o-x6_5YfMv/view?usp=sharing
import onnxruntime as rt
sess = rt.InferenceSession('./model_hack.onnx')
print(sess.get_inputs()[0].name)
print(sess.get_outputs()[0].name)
res = sess.run(['t_output'], {'f_input': [0]})
print(bytes(res[0]))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment