-
Notifications
You must be signed in to change notification settings - Fork 70
Description
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!