nikigoli commited on
Commit
362146f
·
verified ·
1 Parent(s): f753e98

More debugging print statements

Browse files
models/GroundingDINO/groundingdino.py CHANGED
@@ -397,6 +397,7 @@ class GroundingDINO(nn.Module):
397
  dictionnaries containing the two above keys for each decoder layer.
398
  """
399
 
 
400
  if targets is None:
401
  captions = kw["captions"]
402
  else:
@@ -672,6 +673,7 @@ class GroundingDINO(nn.Module):
672
  # outputs['one_hot'].shape
673
  # torch.Size([4, 900, 256])
674
 
 
675
  return out
676
 
677
  @torch.jit.unused
 
397
  dictionnaries containing the two above keys for each decoder layer.
398
  """
399
 
400
+ print("inside forward")
401
  if targets is None:
402
  captions = kw["captions"]
403
  else:
 
673
  # outputs['one_hot'].shape
674
  # torch.Size([4, 900, 256])
675
 
676
+ print("returning out")
677
  return out
678
 
679
  @torch.jit.unused