- 博客(1)
- 收藏
- 关注
原创 RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be
训练yolov8时遇到该问题,debug时conv.py出现return F.conv2d(input, weight, bias, self.stride,报错。在forward里添加一行,输入(input)张量的数据类型转换为与权重(weight)张量相同的数据类型.
2024-03-26 19:53:33 433 4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人