李博士您好,我在测试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 的路径换成自己的红外图像路径,请问是还需要修改其他地方吗?
李博士您好,我在测试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 的路径换成自己的红外图像路径,请问是还需要修改其他地方吗?![1](https://github.com/hli1221/imagefusion-rfn-nest/assets/124424070/e07fc7ea-ba84-4ea5-84a0-017c3465c441)