AttributeError ‘torch.Size‘ object has no attribute ‘numel‘

最新推荐文章于 2023-04-22 15:11:44 发布

pan_mlpan

最新推荐文章于 2023-04-22 15:11:44 发布

阅读量2.2k

点赞数 1

分类专栏：解决办法文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/weixin_42272869/article/details/123774760

版权

34 篇文章 0 订阅

订阅专栏

AttributeError ‘torch.Size’ object has no attribute ‘numel’

使用 torchstat 来统计神经网络中的参数量的时候，出现以下错误：

请添加图片描述

出错描述：使用torchstat 的stat方法报错 AttributeError ‘torch.Size’ object has no attribute 'numel’

出错原因：版本不一致

当时是直接 pip install torchstat 安装的 torchstat 包，没有考虑到目前环境中 pytorch 的版本，而上述问题就是因为安装的torchstat版本不支持已安装的pytorch版本

Pytorch 的版本是 0.4.1 ， torchstat 为0.0.6版本才支持，需要更换版本信息

# 使用该命令安装的是当前最新版本0.0.7的torchstat

pip install torchstat

# 使用该命令安装0.0.6版本的torchstat工具解决了文中的报错

pip install torchstat==0.0.6

更换完成之后，再执行即可！

具体的版本支持情况，可以查看github网站，传送门 torchStat github

关注