nikigoli commited on
Commit
64980f3
1 Parent(s): a30d933

More debugging printing

Browse files
models/GroundingDINO/groundingdino.py CHANGED
@@ -598,6 +598,7 @@ class GroundingDINO(nn.Module):
598
  poss.append(pos_l)
599
 
600
  input_query_bbox = input_query_label = attn_mask = dn_meta = None
 
601
  hs, reference, hs_enc, ref_enc, init_box_proposal = self.transformer(
602
  srcs, masks, input_query_bbox, poss, input_query_label, attn_mask, text_dict
603
  )
 
598
  poss.append(pos_l)
599
 
600
  input_query_bbox = input_query_label = attn_mask = dn_meta = None
601
+ print("passing info through transformer")
602
  hs, reference, hs_enc, ref_enc, init_box_proposal = self.transformer(
603
  srcs, masks, input_query_bbox, poss, input_query_label, attn_mask, text_dict
604
  )