出现的问题:

在深度学习模型使用中,再查看模型的结构参数量、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

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐