使用torchsummary的時(shí)候，出現(xiàn)以下錯(cuò)誤：

（涉及板塊：

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))

錯(cuò)誤：

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

?這是由于模型和summary一個(gè)是cpu版本一個(gè)是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')

總結(jié)

以上是生活随笔為你收集整理的错误处理：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。