关于使用PSA模块报错
作者: nlper01创建于 2022年9月12日更新于 2023年9月19日
Boss, 我在使用`PSA.py`这个模块时，总是报错`RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same`， 我已经把这个模块`.cuda()`了
内容来源: xmu-xiaoma666/External-Attention-pytorch