Hi there,
I’m trying to write a custom nms operator for TorchScript. The original code is here: https://github.com/lucastabelini/LaneATT/tree/main/lib/nms 5.
My code for building nms operator is here: https://github.com/hthoai/LaneATT/tree/torchscript/lib/nms_torchscript 6
I caught this error:

image1393×371 45.6 KB

Is there anyone know how to fix it?!