We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1c0b10c commit 46b204fCopy full SHA for 46b204f
timm/models/dependencyvit.py
@@ -77,7 +77,7 @@ def forward(self, in_tuple: Tuple[torch.Tensor, torch.Tensor]) -> Tuple[torch.Te
77
78
p = (self.head_selector(x) / self.head_selector_temperature).softmax(dim=-1)
79
p = p.transpose(-2, -1).reshape(B, self.num_heads, 1, N)
80
- print(m)
+
81
m = m * self.message_controller(x).sigmoid().reshape(B, 1, 1, N)
82
83
q = q * self.scale
0 commit comments