使用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的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。