使用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里面的红框中的文件修改为：
问题解决，能够正常打印参数。
打印结果：
旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。
更多推荐
30
0- 0
已为社区贡献4条内容
所有评论(0)