@tom Thanks. It is working for me now in torchscript as well. But, the NMS now fails in the TRTserver.
Unknown builtin op: torchvision::nms.
Could not find any similar ops to torchvision::nms. This op may not exist or may not be currently supported in TorchScript.