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

在运用自己的神经网络测试一张图片的时候出现的问题。

错误提示

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be..._第1张图片

 我的输入为一张图片,类型为tensor数据类型

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be..._第2张图片

错误原因就是输入的数据类型与网络模型中的权重类型不匹配。是因为之前在测试网络的时候使用了

device = torch.device("cpu")更换了设备环境,所以在代码前加入下面两行代码,改变输入的运行环境。
device = torch.device('cuda')
image = image.to(device)

类似的,你可以将上面代码中的image换成你输入的变量名称。

 最后成功运行

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be..._第3张图片

 

你可能感兴趣的:(神经网络,深度学习,人工智能,python,pytorch)