and when the function tests the inference with the sample input and converted model i get:
Traceback (most recent call last):
File "trt_convert_yolo.py", line 25, in <module>
model_trt = torch2trt.torch2trt(model_w, [sample_data])
File "/home/shinkei/.virtualenvs/dev/lib/python3.6/site-packages/torch2trt-0.4.0-py3.6.egg/torch2trt/torch2trt.py", line 739, in torch2trt
outputs = module(*inputs)
File "/home/shinkei/.virtualenvs/dev/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1120, in _call_impl
result = forward_call(*input, **kwargs)
File "trt_convert_yolo.py", line 20, in forward
return self.model(x, False)
File "/home/shinkei/.virtualenvs/dev/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1120, in _call_impl
result = forward_call(*input, **kwargs)
File "/home/shinkei/project/CV_pipeline/object_detection/models/yolo.py", line 319, in forward
return self.forward_once(x, profile) # single-scale inference, train
File "/home/shinkei/project/CV_pipeline/object_detection/models/yolo.py", line 345, in forward_once
x = m(x) # run
File "/home/shinkei/.virtualenvs/dev/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1120, in _call_impl
result = forward_call(*input, **kwargs)
File "/home/shinkei/project/CV_pipeline/object_detection/models/yolo.py", line 99, in forward
y[..., 0:2] = (y[..., 0:2] * 2. - 0.5 + self.grid[i]) * self.stride[i] # xy
File "/home/shinkei/.virtualenvs/dev/lib/python3.6/site-packages/torch2trt-0.4.0-py3.6.egg/torch2trt/torch2trt.py", line 310, in wrapper
converter["converter"](ctx)
File "/home/shinkei/.virtualenvs/dev/lib/python3.6/site-packages/torch2trt-0.4.0-py3.6.egg/torch2trt/converters/getitem.py", line 30, in convert_tensor_getitem
input_trt = input._trt
AttributeError: 'Tensor' object has no attribute '_trt'
I suspect it's some missing layers that didn't make it during the conversion process since they were unsupported, but I have not been able to figure out exactly which layers are causing this. Can someone help me on this?
Hi,
I've been trying to convert a yolov7 model to tensorrt format but I get a bunch of the following printouts"
and when the function tests the inference with the sample input and converted model i get:
I suspect it's some missing layers that didn't make it during the conversion process since they were unsupported, but I have not been able to figure out exactly which layers are causing this. Can someone help me on this?