fix cls_x and bbox_x is possibly unbound (#10991)

2026-09-19 10:02:28 +08:00 · 2023-09-25 17:15:09 +08:00
parent 8a52c99ad8
commit 2751cb3a11
1 changed files with 2 additions and 0 deletions
@@ -90,11 +90,13 @@ class TableMasterHead(nn.Layer):
            x = layer(x, feature, src_mask, tgt_mask)

        # cls head
+        cls_x = x
        for layer in self.cls_layer:
            cls_x = layer(x, feature, src_mask, tgt_mask)
        cls_x = self.norm(cls_x)

        # bbox head
+        bbox_x = x
        for layer in self.bbox_layer:
            bbox_x = layer(x, feature, src_mask, tgt_mask)
        bbox_x = self.norm(bbox_x)