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

在使用可视化工具

from pytorch_cnn_visualizations import cnn_layer_visualization

可视化卷积网络时，

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

原因是cpu和cdua 使用的不一致

 

在本问题中，自己的model一直使用的是cdua，但是该可视化工具使用的是cpu，因此使用，将load model前，将device 改成cpu

完整代码如下

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
device = torch.device("cpu")
print(device)
model = torch.load('output/model_C_dataAugmentation_best_model.pt')
model = model.to(device)
visualize_conv(model)

如果是想都转成cdua，对于神经网络则使用：

model = MyModel()
model.cuda()