RuntimeError: expected scalar type Half but found Float
起因:将CCNet的十字交叉注意力模块移植到YOLOv5中。经过:在注意力模块中,会有较多的矩阵运算,在训练时出现了cuda和cup类型的冲突(另一篇我写的文章);而在验证时出现了上述错误。出错的代码如下:#[b1*w1,c1,h1]->[b1,w1,c1,h1]->[b1,c1,h1,w1]out_H=torch.bmm(value_H,att_H.permute(0,2,1)).view(b1