错误处理: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的主要内容,如果未能解决你的问题,请参考以下文章
错误处理: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