阅读背景:

RuntimeError: expected scalar type Half but found Float

来源:互联网 

起因:将CCNet的十字交叉注意力模块移植到YOLOv5中。

经过:在注意力模块中,会有较多的矩阵运算,在训练时出现了cuda和cup类型的冲突(另一篇我写的文章);而在验证时出现了上述错误。经过:在注意力模块中,会有较多的矩阵运算




你的当前访问异常,请进行认证后继续阅读剩余内容。

分享到: