Skip to content

Commit 11e8739

Browse files
committed
Update dependencyvit.py
1 parent aac764c commit 11e8739

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/dependencyvit.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def forward(self, x: torch.Tensor, m: torch.Tensor) -> Tuple[torch.Tensor, torch
120120
#prune_mask = attn.sum(1).abs().sum(-1)
121121
#prune_mask = attn.abs().sum((1, -1))
122122
#prune_mask = m.reshape(B, N)
123-
purne_mask = m.detach().reshape(B, N)
123+
prune_mask = m.detach().reshape(B, N)
124124

125125
x = self.proj(x)
126126
x = self.proj_drop(x)

0 commit comments

Comments
 (0)