fix infer layer (#8162)

4383428f · Wenyu · GitHub · 77333393 · 4383428f
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

ppdet/modeling/transformers/rtdetr_transformer.py ppdet/modeling/transformers/rtdetr_transformer.py +1 -0

未找到文件。
--- a/ppdet/modeling/transformers/rtdetr_transformer.py
+++ b/ppdet/modeling/transformers/rtdetr_transformer.py
@@ -245,6 +245,7 @@ class TransformerDecoder(nn.Layer):
            elif i == self.eval_idx:
                dec_out_logits.append(score_head[i](output))
                dec_out_bboxes.append(inter_ref_bbox)
+                break

            ref_points = inter_ref_bbox
            ref_points_detach = inter_ref_bbox.detach(