From 4383428f3e9d316622032cadea2eaaa07d360b18 Mon Sep 17 00:00:00 2001 From: Wenyu Date: Thu, 4 May 2023 10:46:26 +0800 Subject: [PATCH] fix infer layer (#8162) --- ppdet/modeling/transformers/rtdetr_transformer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ppdet/modeling/transformers/rtdetr_transformer.py b/ppdet/modeling/transformers/rtdetr_transformer.py index 672590edf..0d4b108a3 100644 --- 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( -- GitLab