可以使用thop工具直接统计模型的 Flops and Parameters吗？

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.

GNU Lesser General Public License v3.0

5.2k stars 1.27k forks source link

Open GuoxiongZhang opened 2 years ago

GuoxiongZhang commented 2 years ago

还有就是我想问一下你的FPS统计是基于工具还是手动求取时间平均统计的？我在2080Ti上训练了B3模型（-c=3)，输入大小为1280*384，然后batch size为4时平均时间为70毫秒左右。 thop统计结果：参数量：11.91M 计算量：32.19Gflops

zylo117 commented 2 years ago

跑10次求平均