报错如下：

Traceback (most recent call last):

  File "quantize_and_frozen_v1.py", line 336, in <module>

    detect()

  File "quantize_and_frozen_v1.py", line 225, in detect

    pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)

  File "/home/cm101/xl/cambricon/MLU270/torch2cambricon/yolov5-v3.1/yolov5-3.1/utils/general.py", line 661, in non_max_suppression

    i = torch.ops.torchvision.nms(boxes, scores, iou_thres)

  File "/torch/venv3/pytorch/lib/python3.6/site-packages/torch/_ops.py", line 61, in __getattr__

    op = torch._C._jit_get_operation(qualified_op_name)

RuntimeError: No such operator torchvision::nms