计算量和参数量
todesti2 opened this issue · comments
我已经阅读过所有有关的issues和文档说明,但还是跑不出来
请问有人跑出来了吗,能给我一个指令吗?
这是我的指令和结果
`(paddlepaddle) root@autodl-container-6b1911bd52-d696bc56:~/RT-DETR/rtdetr_paddle# python flops.py
WARNING: OMP_NUM_THREADS set to 12, not 1. The computation speed will not be optimized if you use data parallel. It will fail if this PaddlePaddle binary is compiled with OpenBlas since OpenBlas does not support multi-threads.
PLEASE USE OMP_NUM_THREADS WISELY.
W1201 09:06:51.720057 1005 gpu_resources.cc:119] Please NOTE: device: 0, GPU Compute Capability: 8.6, Driver API Version: 11.7, Runtime API Version: 11.2
W1201 09:06:51.723688 1005 gpu_resources.cc:149] device: 0, cuDNN Version: 8.1.
Traceback (most recent call last):
File "flops.py", line 14, in
paddle.flops(model, None, blob, custom_ops=None, print_detail=False)
TypeError: flops() got multiple values for argument 'custom_ops'`
我这边按照流程操作没问题的 另外我看其他同学也跑出来了, 你这边应该是文档里第一步没改对
我这边按照流程操作没问题的 另外我看其他同学也跑出来了, 你这边应该是文档里第一步没改对
我想得到的是:
我对RT-DETR添加tricks后训练出来的模型 的参数量和计算量
步骤:
用第一步的代码替换对应文件中的flops函数。【只替换,其他什么都没改】
接着创建文件get.py,放入第二步的代码并执行【只改将config改成绝对路径】
问题:
Traceback (most recent call last): File "get.py", line 14, in <module> paddle.flops(model, None, blob, custom_ops=None, print_detail=False) TypeError: flops() got multiple values for argument 'custom_ops'
不管尝试多少次都是这个问题,困扰太久实在没有法子了,只能来求助您……请帮助我❤
第一步怎么替换的 替换之后的文件upload看下
- 你在flops函数里随便加一行 比如在30行 + print('xxx') ;看看走这个函数了嘛
- get.py不需要放到hapi下面 在你工作目录随便找个地方放
- 确认下你用的是那个python以及对应的paddle安装位置 (一般
paddle
在对应python环境的site-packages
下)
which python