Skip to content

Commit 46b204f

Browse files
committed
Update dependencyvit.py
1 parent 1c0b10c commit 46b204f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/dependencyvit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def forward(self, in_tuple: Tuple[torch.Tensor, torch.Tensor]) -> Tuple[torch.Te
7777

7878
p = (self.head_selector(x) / self.head_selector_temperature).softmax(dim=-1)
7979
p = p.transpose(-2, -1).reshape(B, self.num_heads, 1, N)
80-
print(m)
80+
8181
m = m * self.message_controller(x).sigmoid().reshape(B, 1, 1, N)
8282

8383
q = q * self.scale

0 commit comments

Comments
 (0)