YOLOv3-Torch2TRT icon indicating copy to clipboard operation
YOLOv3-Torch2TRT copied to clipboard

[TensorRT] ERROR: (Unnamed Layer* 203) [Concatenation]: all concat input tensors must have the same number of dimensions, but mismatch at input 1. Input 0 shape: [1,256,26,26], Input 1 shape: [512,26,26]

Open haixiansheng opened this issue 4 years ago • 0 comments

[TensorRT] ERROR: (Unnamed Layer* 203) [Concatenation]: all concat input tensors must have the same number of dimensions, but mismatch at input 1. Input 0 shape: [1,256,26,26], Input 1 shape: [512,26,26] Traceback (most recent call last): File "detect.py", line 128, in model_trt = torch2trt(model_backbone, [x]) File "/usr/local/lib/python3.6/dist-packages/torch2trt/torch2trt.py", line 436, in torch2trt outputs = module(*inputs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in _call_impl result = self.forward(*input, **kwargs) File "/srv/YOLOv3-Torch2TRT/models.py", line 298, in forward x = module(x) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in call_impl result = self.forward(*input, **kwargs) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/container.py", line 117, in forward input = module(input) File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 722, in call_impl result = self.forward(*input, **kwargs) File "/usr/local/lib/python3.6/dist-packages/torch2trt/torch2trt.py", line 218, in wrapper converter"converter" File "/usr/local/lib/python3.6/dist-packages/torch2trt/converters/Conv.py", line 10, in convert_Conv_trt7 input_trt = trt(ctx.network, input) File "/usr/local/lib/python3.6/dist-packages/torch2trt/torch2trt.py", line 132, in trt t._trt.shape ValueError: len() should return >= 0

pytoch: 1.4.0 tensorRT: 7.0.0 报以上错误,求助小伙伴,感激不尽

haixiansheng avatar Aug 13 '20 08:08 haixiansheng