Skip to content

Expose model inputs and outputs number in AI.MODELGET #529

New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Closed
2 tasks
DvirDukhan opened this issue Dec 3, 2020 · 1 comment · Fixed by #552
Closed
2 tasks

Expose model inputs and outputs number in AI.MODELGET #529

DvirDukhan opened this issue Dec 3, 2020 · 1 comment · Fixed by #552

Comments

@DvirDukhan
Copy link
Collaborator

DvirDukhan commented Dec 3, 2020

As INPUTS and OUTPUTS are required only for TF models in the AI.MODELSET command, we want to allow investigate model's input and outputs:

  • ONNX:
 size_t n_input_nodes;
 ort->SessionGetInputCount(session, &n_input_nodes);
 size_t n_output_nodes;
 ort->SessionGetOutputCount(session, &n_output_nodes);
  • Torch - User defined
@nishantgeorge
Copy link
Contributor

I think AI.MODELGET might be more appropriate since it already returns information about the model's inputs, and AI.INFO does not.

@DvirDukhan DvirDukhan changed the title Expose model inputs and outputs number in AI.INFO Expose model inputs and outputs number in AI.MODELGET Dec 6, 2020
# for free to join this conversation on GitHub. Already have an account? # to comment
Projects
None yet
2 participants