[paddledet][深度学习][原创]paddledet打印出FLops正确方法

Posted 2022-12-19 FL1623863129

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了[paddledet][深度学习][原创]paddledet打印出FLops正确方法相关的知识，希望对你有一定的参考价值。

第一步：去PaddleDetection\\configs\\runtime.yaml将print_flops设置true

在paddle环境安装paddleslim，我是paddlepaddle-gpu==2.3.2但是我安装paddleslim==2.4.0会报错

ImportError: cannot import name '_legacy_C_ops' from 'paddle'

然后我换成2.3.4版本就好了pip install paddleslim==2.3.4

最后python tools/infer.py执行命令就可以看到flops了，最后说明一下我的环境

paddledet==2.5.0

paddlepaddle-gpu==2.3.2

paddleslim==2.3.4

以上是关于[paddledet][深度学习][原创]paddledet打印出FLops正确方法的主要内容，如果未能解决你的问题，请参考以下文章