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
|