Spaces:
Running
on
T4
Running
on
T4
Printing layer index while debugging
Browse files
models/GroundingDINO/groundingdino.py
CHANGED
@@ -576,6 +576,7 @@ class GroundingDINO(nn.Module):
|
|
576 |
srcs = []
|
577 |
masks = []
|
578 |
for l, feat in enumerate(features):
|
|
|
579 |
src, mask = feat.decompose()
|
580 |
srcs.append(self.input_proj[l](src))
|
581 |
masks.append(mask)
|
|
|
576 |
srcs = []
|
577 |
masks = []
|
578 |
for l, feat in enumerate(features):
|
579 |
+
print("l: " + str(l))
|
580 |
src, mask = feat.decompose()
|
581 |
srcs.append(self.input_proj[l](src))
|
582 |
masks.append(mask)
|