训练时关于attention.py中的nms_crop计算时间很长

Open wlk788 opened this issue 2 years ago • 2 comments

您好，感谢您的工作！我在训练的时候发现GPU利用率只有20-40%，然后调试时我发现nms_crop这个模块的计算时间很长，在2080ti下几乎要花费1s的时间，请问这是正常的吗？

Dec 18 '23 09:12 wlk788

您好，感谢您对我们工作的支持。这一部分模块可能会占用比较大的CPU线程消耗，在CPU计算量不足的情况下，每次迭代的part搜索会导致速度变慢。由于当时使用服务器进行计算，并未发现明显速度影响。您可以尝试调整部件个数来极大缩短计算时间。：）

Dec 21 '23 08:12 zhao1f

好的，感谢您的回复！

Dec 21 '23 08:12 wlk788