训练yolov8时遇到该问题,debug时conv.py出现return F.conv2d(input, weight, bias, self.stride,报错
解决方法:
在forward里添加一行,输入(input)张量的数据类型转换为与权重(weight)张量相同的数据类型.
input = input.to(self.weight.dtype)
如下图所示:
修改后问题解决!
训练yolov8时遇到该问题,debug时conv.py出现return F.conv2d(input, weight, bias, self.stride,报错
解决方法:
在forward里添加一行,输入(input)张量的数据类型转换为与权重(weight)张量相同的数据类型.
input = input.to(self.weight.dtype)
如下图所示:
修改后问题解决!