解决使用torchstat时报错“AttributeError: module ‘numpy‘ has no attribute ‘long‘”等问题

背景

首先直接使用pip install torchstat安装。
使用torchstat查看模型参数和flops：

from torchstat import stat
stat(model.to('cpu'), (2, 32, 32))  # 这里第二个参数取决于自己的模型输入大小
1
2

报错1

运行报错如下：

核心错误为：
“AttributeError: module ‘numpy’ has no attribute ‘long’”

报错1解决：修改包的源代码

此处修改思路为：直接打开下图标识出来的这个文件夹中的.py文件：


把第65行这里的long改成longlong

重新load kernel，再次运行，上面那个报错解决了，但出现了新的报错。

报错2

新的报错为：
“AttributeError: ‘DataFrame‘ object has no attribute ‘append‘”

报错2解决：降低pandas版本至1.3.4

这次不能按照上面的思路直接改包文件的代码，而是需要降低pandas的版本。
查看pandas的版本：
进入目标环境的python后输入如下内容：

import pandas
pandas.__version__
1
2

可得：

退出python（输入exit()），在目标环境中输入：

pip install pandas==1.3.4 --user
1

等待安装成功。
再次进入python检查此时的pandas版本：

成功。

debug效果

再次运行，此时就可以正常输出模型结构和相关参数了！