模型固化出现问题

[kycwx]

❔Question

感谢您的工作，效果非常好！！！
但我在固化您的模型以便在c++中进行调用的时候，遇到了一些问题
在使用 torch.jit.trace() 时候发生了问题
ts = torch.jit.trace(model, out)
ts.save("pspv5s.torchscript.pt")
提示我:List inputs to traced functions must have consistent element type. Found Tuple[Tensor, List[Tensor]] and Tensor

能否请求您的帮助？

Additional context

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.