关于test_40pairs的问题

hli1221 / imagefusion-rfn-nest

RFN-Nest(Information Fusion, 2021, Highly Cited Paper) - PyTorch =1.5，Python=3.7

108 stars 18 forks source link

关于test_40pairs的问题 #16

Open daidaidaihaohaohao opened 3 months ago

daidaidaihaohaohao commented 3 months ago

李博士您好，我在测试test_40pairs.py文件时使用自己的数据集时显示RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor。我是将test_path 的路径换成自己的红外图像路径，请问是还需要修改其他地方吗？