Update linea/models/linea/decoder.py

linea/models/linea/decoder.py

@@ -196,7 +196,7 @@ class TransformerDecoder(nn.Module):
         self.integral = Integral(self.reg_max)
 
         # two stage
-        self.enc_out_bbox_embed = copy.deepcopy(_enc_bbox_embed)
+        # self.enc_out_bbox_embed = copy.deepcopy(_enc_bbox_embed)
         # self.enc_out_class_embed = copy.deepcopy(_class_embed)
         self.aux_loss = aux_loss
 
@@ -261,7 +261,7 @@ class TransformerDecoder(nn.Module):
             )
 
             if layer_id == 0:
-            	pre_bboxes = torch.sigmoid(self.enc_out_bbox_embed(output) + inverse_sigmoid(ref_points_detach))
+            	pre_bboxes =  ref_points_detach
             	pre_scores = self.class_embed[0](output)
             	ref_points_initial = pre_bboxes.detach()