Open lucasjinreal opened 3 years ago
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.
I modified the output but I still got error AttributeError: 'Tensor' object has no attribute '_trt' in this function def mark_outputs(self, torchoutputs, names=None): if names is None: names = ['output%d' % i for i in range(len(torch_outputs))] ....... self.network.mark_output(trt_tensor)
I got this error when using list as output: