Spaces:
Running
on
T4
Running
on
T4
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
|