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

原创已于 2023-07-17 17:33:01 修改 · 351 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#神经网络

收录于

于 2023-07-17 16:52:17 首次发布

深度学习543 专栏收录该内容

10 篇文章

订阅专栏

在尝试测试MobilenetV3网络时遇到了RuntimeError，问题出在输入类型torch.cuda.FloatTensor和权重类型torch.FloatTensor不匹配。当使用GPU时，需要将数据通过cuda()转移到GPU上进行计算；若使用CPU，则无此问题。可以通过调整数据输入方式来解决这个问题。

【小白学习笔记】测试mobilenetv3网络如下错误：

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

    在测试打印网络参数情况时，用如下片段。发现是由于电脑有GPU，数据需要送入cuda()进行运算；如果电脑用CPU则正常。

if __name__ == '__main__':
    #net = MobileNetV3_Small().train().cuda() #GPU训练
    net = MobileNetV3_Small()                 #CPU训练
    summary(net, (3, 224, 224))