使用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里面的红框中的文件修改为:

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

打印结果:

     

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值