Skip to content

TypeError: list indices must be integers or slices, not list #43

@LucQueen

Description

@LucQueen

File "/opt/conda/envs/torch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) File "/opt/conda/envs/torch/lib/python3.7/site-packages/onnx2pytorch-0.4.1-py3.7.egg/onnx2pytorch/convert/model.py", line 300, in forward File "/opt/conda/envs/torch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) File "/opt/conda/envs/torch/lib/python3.7/site-packages/onnx2pytorch-0.4.1-py3.7.egg/onnx2pytorch/operations/gather.py", line 15, in forward TypeError: list indices must be integers or slices, not list

when convert dlrm onnx to pytorch model, I got this error, and I debug the data shape is torch.Size([4, 2]) selection shape is torch.Size([1, 128]) , the data and the selection shape is mismatch, I find that dlrm has dynamic axes.
any idea to solve this error? thanks very much!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions