RuntimeError: Only tensors, lists, tuples of tensors, or dictionary of tensors can be output from traced functions.

xingyizhou / CenterNet

Object detection, 3D detection, and pose estimation using center point detection:

MIT License

7.3k stars 1.93k forks source link

RuntimeError: Only tensors, lists, tuples of tensors, or dictionary of tensors can be output from traced functions. #1035

Open chjXu opened 1 year ago

chjXu commented 1 year ago

when i try to save the ddd_3dop.pth as the ddd_3dop.pt for libtorch, there will be occured the error as "RuntimeError: Only tensors, lists, tuples of tensors, or dictionary of tensors can be output from traced functions.". How to sovled it? please help me, thanks.