RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

Linfeng-Tang / SeAFusion

The code of " Image fusion in the loop of high-level vision tasks: A semantic-aware real-time infrared and visible image fusion network"

MIT License

184 stars 34 forks source link

Open cheeenzq opened 2 years ago

cheeenzq commented 2 years ago

请问为什么会出现这个问题，感谢。

Linfeng-Tang commented 1 year ago

应该是你的模型没有加载到CUDA上导致你可以检查一下模型加载和数据加载的程序看看。