Hi, how to solve it with list output?

[lucasjinreal]

I got this error when using list as output:

line 408, in mark_outputs
    trt_tensor = torch_output._trt
AttributeError: 'list' object has no attribute '_trt'

[grimoire]

What is the output structure of your model?
The convertor only accept tensor or list of tensors as output. Nested list is not convertable as the final output.

[LeoRainly]

I modified the output but I still got error
AttributeError: 'Tensor' object has no attribute 'trt'
in this function
def mark_outputs(self, torch_outputs, names=None):
if names is None:
names = ['output%d' % i for i in range(len(torch_outputs))]
.......
self.network.mark_output(trt_tensor)