RuntimeError: "upsample_nearest2d_channels_last" not implemented for 'Half'

[charley12009]

我上網查了，好像跟torch版本有關，但現在只能裝2.x版而已，請問一下該如何解決?
Traceback (most recent call last):
File "c:\yolov5_deepsort_infer\Yolov5-deepsort-inference\demo.py", line 58, in
main()
File "c:\yolov5_deepsort_infer\Yolov5-deepsort-inference\demo.py", line 32, in main
result = det.feedCap(im, func_status)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "c:\yolov5_deepsort_infer\Yolov5-deepsort-inference\utils\BaseDetector.py", line 35, in feedCap
im, faces, face_bboxes = update_tracker(self, im)
^^^^^^^^^^^^^^^^^^^^^^^^
File "c:\yolov5_deepsort_infer\Yolov5-deepsort-inference\tracker.py", line 42, in update_tracker
_, bboxes = target_detector.detect(image)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "c:\yolov5_deepsort_infer\Yolov5-deepsort-inference\AIDetector_pytorch.py", line 47, in detect
pred = self.m(img, augment=False)[0]
^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\yolov5_deepsort_infer\yolov5_infer\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\yolov5_deepsort_infer\yolov5_infer\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "c:\yolov5_deepsort_infer\Yolov5-deepsort-inference\models\yolo.py", line 112, in forward
return self.forward_once(x, profile) # single-scale inference, train
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "c:\yolov5_deepsort_infer\Yolov5-deepsort-inference\models\yolo.py", line 128, in forward_once
x = m(x) # run
^^^^
File "C:\yolov5_deepsort_infer\yolov5_infer\Lib\site-packages\torch\nn\modules\module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\yolov5_deepsort_infer\yolov5_infer\Lib\site-packages\torch\nn\modules\module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\yolov5_deepsort_infer\yolov5_infer\Lib\site-packages\torch\nn\modules\upsampling.py", line 157, in forward
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\yolov5_deepsort_infer\yolov5_infer\Lib\site-packages\torch\nn\functional.py", line 4001, in interpolate
return torch._C._nn.upsample_nearest2d(input, output_size, scale_factors)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: "upsample_nearest2d_channels_last" not implemented for 'Half'