diff --git a/timm/models/dependencyvit.py b/timm/models/dependencyvit.py index 45c1d3d24..92fd8612a 100644 --- a/timm/models/dependencyvit.py +++ b/timm/models/dependencyvit.py @@ -120,7 +120,7 @@ def forward(self, x: torch.Tensor, m: torch.Tensor) -> Tuple[torch.Tensor, torch #prune_mask = attn.sum(1).abs().sum(-1) #prune_mask = attn.abs().sum((1, -1)) #prune_mask = m.reshape(B, N) - purne_mask = m.detach().reshape(B, N) + prune_mask = m.detach().reshape(B, N) x = self.proj(x) x = self.proj_drop(x)