运行模块时报错请教

YimianDai / open-aff

code and trained models for "Attentional Feature Fusion"

729 stars 95 forks source link

Open GeniusLittleFlyingPig opened 1 year ago

GeniusLittleFlyingPig commented 1 year ago

您好，我在将您提供的模块添加进网络后，出现了如下报错，请问这要怎么解决呢？ RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

YimianDai commented 1 year ago

请以 mxnet 代码为准，pytorch 代码没有经过测试