当前位置:   article > 正文

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

runtimeerror: input type (torch.floattensor) and weight type (torch.cuda.flo

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

出现这个问题是因为输入模型的tensor加载在cpu中,而模型却加载在cuda上。

解决方法:将输入tensor加载到cuda中,或者把模型加载到cpu

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

model = model.to(device)
img = img.to(device)

output = model(img)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

或者:

model = model.cuda()
img = img.cuda()
  • 1
  • 2
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/103340
推荐阅读
相关标签
  

闽ICP备14008679号