Skip to content

Commit 6a373ac

Browse files
committed
Update dependencyvit.py
1 parent b662b52 commit 6a373ac

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

timm/models/dependencyvit.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,6 +154,7 @@ def forward(self, in_tuple: Tuple[torch.Tensor, torch.Tensor]) -> Tuple[torch.Te
154154
x_new, m = self.attn((self.norm1(x), m))
155155
x = x + self.drop_path1(self.ls1(x_new))
156156
x = x + self.drop_path2(self.ls2(self.mlp(self.norm2(x))))
157+
print((x, m))
157158
return (x, m)
158159

159160
# FIXME lite model variants

0 commit comments

Comments
 (0)