未验证 提交 4383428f 编写于 作者: W Wenyu 提交者: GitHub

fix infer layer (#8162)

上级 77333393
...@@ -245,6 +245,7 @@ class TransformerDecoder(nn.Layer): ...@@ -245,6 +245,7 @@ class TransformerDecoder(nn.Layer):
elif i == self.eval_idx: elif i == self.eval_idx:
dec_out_logits.append(score_head[i](output)) dec_out_logits.append(score_head[i](output))
dec_out_bboxes.append(inter_ref_bbox) dec_out_bboxes.append(inter_ref_bbox)
break
ref_points = inter_ref_bbox ref_points = inter_ref_bbox
ref_points_detach = inter_ref_bbox.detach( ref_points_detach = inter_ref_bbox.detach(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册