Fail to run the onnx demo
Open
Nobody has claimed this yet.
- Dominant language
- Jupyter Notebook
- Stars
- 9.8k
- Forks
- 1.6k
- PR merge metrics
- No merged PRs in 30d
Description
I try to run the onnx demo:
https://github.com/onnx/models/blob/master/vision/classification/imagenet_inference.ipynb
with model squeezenet, but i got error:
ValueError: You created Module with Module(..., data_names=['data']) but input with name 'data' is not found in symbol.list_arguments(). Did you mean one of:
data_0
conv1_w_0
conv1_b_0
fire2/squeeze1x1_w_0
...
I find the input node name is data, what should i do?
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the linked vision/classification/imagenet_inference.ipynb notebook and reproduce the SqueezeNet run using the reported error. Inspect how the demo identifies the model input and compare it with the symbol arguments shown in the traceback. Done means the notebook runs successfully with SqueezeNet and no longer reports the missing data input.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- jupyter-notebook
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100