More printing for debugging purposes

models/GroundingDINO/groundingdino.py

@@ -409,6 +409,7 @@ class GroundingDINO(nn.Module):
             samples.device
         )
 
+        print("tokenized text")
         one_hot_token = tokenized
 
         (
@@ -444,6 +445,7 @@ class GroundingDINO(nn.Module):
 
         bert_output = self.bert(**tokenized_for_encoder)  # bs, 195, 768
 
+        print("got bert output")
         encoded_text = self.feat_map(
             bert_output["last_hidden_state"]
         )  # bs, 195, d_model
@@ -492,6 +494,8 @@ class GroundingDINO(nn.Module):
             else:
                 exemplar_tokens = None
 
+            print("got visual exemplar tokens")
+
         else:
             features, poss = self.backbone(samples)
             (h, w) = (
@@ -597,6 +601,8 @@ class GroundingDINO(nn.Module):
             srcs, masks, input_query_bbox, poss, input_query_label, attn_mask, text_dict
         )
 
+        print("passed info through transformer")
+
         # deformable-detr-like anchor update
         outputs_coord_list = []
         for dec_lid, (layer_ref_sig, layer_bbox_embed, layer_hs) in enumerate(