【问题记录】RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should b

最新推荐文章于 2024-09-14 21:53:48 发布

超好的小白

最新推荐文章于 2024-09-14 21:53:48 发布

阅读量124

点赞数 3

分类专栏：深度学习遇到的问题文章标签：学习深度学习

本文链接：https://blog.csdn.net/qq_46110320/article/details/135898579

版权

深度学习遇到的问题专栏收录该内容

7 篇文章 0 订阅

订阅专栏

文章介绍了在使用预训练VGG19时遇到的类型错误，解决方法是将模型移到GPU并调整代码。

摘要由CSDN通过智能技术生成

代码：

# 使用预训练好的VGG19网络为基础网络
model_vgg19 = vgg19(pretrained=True)
## 不使用vgg19网络中后面的AdaptiveAvgPool2d 和Linear层
base_model = model_vgg19.features

summary(base_model, input_size=(3,high,width))

遇到报错：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
报错意思是说，模型和输入数据类型不匹配，模型在GPU上，而输入数据是在CPU上。所以，我们要将模型放在GPU上，修改代码如下即可成功运行：

# 使用预训练好的VGG19网络为基础网络
model_vgg19 = vgg19(pretrained=True)
## 不使用vgg19网络中后面的AdaptiveAvgPool2d 和Linear层
base_model = model_vgg19.features

# 将模型移动到GPU上
base_model = base_model.to(device)

summary(base_model, input_size=(3,high,width))