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
错误原因:
这个错误提示表明输入数据的类型和模型权重的类型不匹配,可能是因为模型权重是在GPU上训练的,而输入数据是在CPU上进行的。文章来源:https://www.toymoban.com/news/detail-739826.html
可以使用model.to('cpu')
将模型权重转移到CPU上,然后再对输入数据进行处理。文章来源地址https://www.toymoban.com/news/detail-739826.html
model_path = '模型路径'
model = torch.load(model_path)
model.to('cpu')
到了这里,关于Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!