使用stat时出现“‘DataFrame‘ object has no attribute ‘append‘”的解决办法

出现的问题:

在深度学习模型使用中,再查看模型的结构参数量、Flops等参数时,要用到stat函数。

比如查看AlexNet模型的命令:

    from torchstat import stat
    import torchvision.models as models
    model = models.AlexNet()
    stat(net, (3, 224, 224))

结果在运行的时候出现错误:

AttributeError: 'DataFrame' object has no attribute 'append'. Did you mean: '_append'?

问题的解决:

网上有人建议降低pandas版本,实际操作后,问题依然存在。

经过分析发现问题是出在显示阶段,应该是在打印报告的时候出错,所以定位到reporter.py程序里面。我的程序地址路径如下,在Anaconda安装目录下的虚拟环境里面。我的虚拟环境是py3.10_cuda11.3。

路径:"D:\Anaconda3\envs\py3.10_cuda11.3\lib\site-packages\torchstat\reporter.py

打开reporter.py后,定位到下图的红框位置。

 将reporter.py里面的红框中的文件修改为:

 

 问题解决,能够正常打印参数。

打印结果:

 https://gitcode.csdn.net/65e937ac1a836825ed78da7d.html?dp_token=eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJpZCI6Mzg3NDA2LCJleHAiOjE3MjkyMzg4MDksImlhdCI6MTcyODYzNDAwOSwidXNlcm5hbWUiOiJ6b3V4aWFvbHYifQ.2pTTVx3RXwN8S8vfceuSOlXvxg4jYpFPSGGmROZ4iTY&spm=1001.2101.3001.6650.17&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Eactivity-17-130554919-blog-130577543.235%5Ev43%5Epc_blog_bottom_relevance_base1&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7Eactivity-17-130554919-blog-130577543.235%5Ev43%5Epc_blog_bottom_relevance_base1

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值