错误处理:RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be

Posted UQI-LIUWJ

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了错误处理:RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be相关的知识,希望对你有一定的参考价值。

使用torchsummary的时候,出现以下错误:

(涉及板块:

python 库整理:Timm(1)_UQI-LIUWJ的博客-CSDN博客

pytorch 笔记:torchsummary_UQI-LIUWJ的博客-CSDN博客

 代码部分

import timm
model = timm.create_model('resnet18', pretrained=True)
model.to_device('cpu')
model.eval()
from torchsummary import summary
summary(model,(3,244,244))

错误:

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

 这是由于模型和summary一个是cpu版本一个是gpu版本

解决方法:(看最后一行)

方法1:

import timm
model = timm.create_model('resnet18', pretrained=True)
model.eval()
from torchsummary import summary
summary(model.cuda(),(3,244,244))

方法2:

import timm
model = timm.create_model('resnet18', pretrained=True)
model.eval()
from torchsummary import summary
summary(model,(3,244,244),device='cpu')

以上是关于错误处理:RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be的主要内容,如果未能解决你的问题,请参考以下文章

导入 PyQt5 时出现 RuntimeError

错误处理:RuntimeError: [enforce fail at ..caffe2serializeinline_container.cc:145] . PytorchStreamRead(代码

PyTorch:RuntimeError:输入、输出和索引必须在当前设备上

Discord.py - 重新启动命令错误 - RuntimeError:事件循环已关闭

“RuntimeError: Expected 4-dimensional input for 4-dimensional weight 32 3 3, but got 3-dimensional i

在PyQt脚本中使用多处理时的RuntimeError和IOError