使用torchsummary时出现AttributeError: ‘list‘ object has no attribute ‘size‘解决方案
目录
1 问题
2 原因
3 解决方案
4 代码
1 问题
使用torchsummary查看网络结构时报错：AttributeError: 'list' object has no attribute 'size'，具体如下图所示。
2 原因
torchsummary的问题，其实还有一个跟它长得很像的库，叫做torch-summary，中间多了一个横杠，它是torchsummary的加强版。这里我们卸掉torchsummary库，安装torch-summary库，上面的问题就可以解决了。
3 解决方案
pip uninstall torchsummary
pip install torch-summary==1.4.44 代码
import torch
from nets.hrnet import get_neck_net
from utils.utils import load_yaml
from torchsummary import summary
import osif __name__ == '__main__':curPath = os.path.dirname(os.path.realpath(__file__))yamlPath = os.path.join(curPath, "config/config_512.yaml")cfg = load_yaml(yamlPath)inputs = torch.rand((2, 3, 512, 512))model = get_neck_net(cfg, is_train=True)summary(model)使用torchsummary时出现AttributeError: ‘list‘ object has no attribute ‘size‘解决方案相关推荐
- keras2onnx转换时出现AttributeError: ‘KerasTensor‘ object has no attribute ‘graph’错误
[问题] keras2onnx转换时出现AttributeError: 'KerasTensor' object has no attribute 'graph'错误 File "D:\co ...
- AttributeError: ‘ChatGLMConfig‘ object has no attribute ‘quantization_bit‘解决方案
大家好,我是爱编程的喵喵.双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中.从事机器学习以及相关的前后端开发工作.曾在阿里云.科大讯飞.CCF等比赛获得多次Top名次.现 ...
- AttributeError: ‘LTP‘ object has no attribute ‘init_dict‘解决方案
大家好,我是爱编程的喵喵.双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中.从事机器学习以及相关的前后端开发工作.曾在阿里云.科大讯飞.CCF等比赛获得多次Top名次.现 ...
- AttributeError: ‘HTMLParser‘ object has no attribute ‘unescape‘解决方案
大家好,我是爱编程的喵喵.双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中.从事机器学习以及相关的前后端开发工作.曾在阿里云.科大讯飞.CCF等比赛获得多次Top名次.喜 ...
- AttributeError: ‘Timestamp‘ object has no attribute ‘dt‘解决方案
大家好,我是爱编程的喵喵.双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中.从事机器学习以及相关的前后端开发工作.曾在阿里云.科大讯飞.CCF等比赛获得多次Top名次.本 ...
- Cython——[AttributeError: ‘MSVCCompiler‘ object has no attribute ‘compiler_so‘]解决方案
问题描述 运行环境 Windows 10 Python3.7 TensorFlow 1.3.0 GitHub文件URL:https://github.com/xiaofengShi/CHINESE-O ...
- AttributeError: ‘str‘ object has no attribute ‘copy
在加载模型时出现 AttributeError: 'str' object has no attribute 'copy' 错误 解决方案: 将 model.load_state_dict(state ...
- AttributeError: ‘Book‘ object has no attribute ‘sheet‘
xlrd读取表时出现AttributeError: 'Book' object has no attribute 'sheet'错误 源码 import xlrd >>> xlsx= ...
- web.py——运行错误【AttributeError: ‘StaticApp‘ object has no attribute ‘directory‘】
问题描述 AttributeError("'StaticApp' object has no attribute 'directory'") Traceback (most rec ...
最新文章
- HDU 2149 Public Sale （博弈）
- Linux strace 跟踪进程信息
- python hmac_Python实现的HMacMD5加密算法示例
- 各安全浏览器如何设2345为主页
- [SPP-NET]Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition
- centos7.4 U盘安装卡在 starting dracut initqueue hook
- 基于NXP i.MX8 Ethercat实时运动控制器设计 Linux+xenomai+igh+rtnet
- 电视android怎么连接手机助手下载,投屏助手app下载 投屏助手(手机投屏到电视) for Android v2.0 安卓版 下载-脚本之家...
- 图形美不胜收，25 个可视化案例，Matplotlib 始终都是数据可视化绕不开的 Python 库
- angular写的移动端模板《二》
- nagios监控系统
- visio2013报临时目录下某些文件找不到（实际有）
- docker进阶：自定义镜像、网络架构（二）
- Flutter混合栈管理
- 文件夹正在使用，无法删除 无法重命名等操作怎么办？
- 华清远见嵌入式Linux就业培训班
- 按键精灵简单的限时找图并点击
- IDEA打包下载超时（已经配置阿里私服，去中央私服下载）
- 可以通过限定ip来限制用户重复登录么_饿了么到底有多“贼”？揭秘它一步步让你下单的套路！...
- Hadoop分布式集群搭建完全详细教程