使用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后,定位到下图的红框位置。

使用stat时出现“‘DataFrame‘ object has no attribute ‘append‘”的解决办法_第1张图片

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

使用stat时出现“‘DataFrame‘ object has no attribute ‘append‘”的解决办法_第2张图片

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

打印结果:

     使用stat时出现“‘DataFrame‘ object has no attribute ‘append‘”的解决办法_第3张图片

 

你可能感兴趣的:(深度学习,pandas,开发语言)