logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

MiXFormer--PyTorch复现

文章:MixFormer: Mixing Features across Windows and Dimensions (CVPR 2022 Oral)原文章的代码实现是在paddlepaddle平台上;我在 @太阳花的小绿豆 的SwinTransformer的基础上更改了一些代码实现的。关键代码如下:model.pyimport torchimport torch.nn as nnimport

#神经网络#pytorch#transformer +1
错误记录: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中。

#python#深度学习#人工智能
到底了