大batch_size和多目标检测的双层for循环导致ATSS算法执行时间长，GPU利用率低，如何提升性能与效率？

[chenrui17]

如题，batch _size = 15， 检测目标类别=6，则atss算法会连续调用90次，通过profile，该部分GPU利用率低，耗时占比训练 30%左右，如何考虑优化，从而提升性能 与GPU利用率呢？

[suekarry]

如题，batch _size = 15， 检测目标类别=6，则atss算法会连续调用90次，通过profile，该部分GPU利用率低，耗时占比训练 30%左右，如何考虑优化，从而提升性能 与GPU利用率呢？

您好，我想问一下，您知道论文里的伪代码在哪个文件里？我找了好久没有找到符合的

[Tangled12]

应该是在 atss_core/modeling/rpn/atss/loss.py
里面的prepare_target是准备计算loss的target，也就是匹配anchor和target

[suekarry]

应该是在 atss_core/modeling/rpn/atss/loss.py 里面的prepare_target是准备计算loss的target，也就是匹配anchor和target

是的！谢谢你~