
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
MiXFormer--PyTorch复现
文章:MixFormer: Mixing Features across Windows and Dimensions (CVPR 2022 Oral)原文章的代码实现是在paddlepaddle平台上;我在 @太阳花的小绿豆 的SwinTransformer的基础上更改了一些代码实现的。关键代码如下:model.pyimport torchimport torch.nn as nnimport
错误记录:RuntimeError: Output 0 of SelectBackward0 is a view and is being modified inplace
RuntimeError: Output 0 of SelectBackward0 is a view and is being modified inplace. This view is the output of a function that returns multiple views. Such functions do not allow the output views to be
RuntimeError: expected scalar type Half but found Float
经过:在注意力模块中,会有较多的矩阵运算,在训练时出现了cuda和cup类型的冲突(另一篇我写的文章);而在验证时出现了上述错误。出错的位置在torch.bmm()处,在这里进行了一次矩阵乘法运算。由于两个数据的类型不同,因此发生冲突。解决方案:仍然是用to()方法,修改数据类型为另一个数据的类型。起因:将CCNet的十字交叉注意力模块移植到YOLOv5中。
到底了







