背景:自己网络模型的结果不是特别好,然后就想多路径出发。从模型参数和处理速度上去与其他模型进行对比,尝试去找到我的模型存在的必要性。故:接触了thop这个第三方安装库,狠狠的踩了一个雷。今天记录一下,也避免大家重复踩雷。
thop的安装问题
在python中安装过第三方库的小伙伴都知道,它的安装命令是:
pip install XXX
。然而针对Thop的安装却有一个大坑。
1)直接使用常用命令
pip install thop
它会默认安装0.0.31.post2005241907
的版本。由于我使用的pytorch环境为1.0.0
,不支持: nn.SyncBatchNorm
(多GPU执行所需的),而该版本下的THop【THOP:0.031】没有对pytorch:1.0.0修复此bug,从而导致运行出错。
2)使用git库安装方式
pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git
第二种方式安装的就会是最新的Thop库的结果!!!超级方便!!!
版权声明:本文为qq_37844044原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。