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)