Hi, I meet some problem about libTorch. When I load my model, I find the code can’t find torch.ops.torchvision.nms. Anybody know how to deal it? Thx~
The error like this:
terminate called after throwing an instance of ‘torch::jit::ErrorReport’
what():
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.
:
File “/home/hookzeng/python-project/pytorch/SSDv3_edition_v2/utils/data_process.py”, line 495
“”"
torch.ops
return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
~~~~~~~~~~~~~~~~~~~~~~~~~ <— HERE