Skip to content

Commit 130e3bb

Browse files
committed
Update dependencyvit.py
1 parent 4e28e34 commit 130e3bb

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
@@ -225,7 +225,7 @@ def __init__(
225225
assert self.prune_ratio * len(self.prune_layers) < 1, "prune_ratio too big, ensure len(prune_layers) * prune_ratio is less than 1"
226226

227227
self.prune_layers = [x-1 for x in self.prune_layers] # convert counting numbers to nn.Sequential indicess
228-
for prune_index, layer in enumerate(prune_layers, 1):
228+
for prune_index, layer in enumerate(self.prune_layers, 1):
229229
self.blocks[layer].token_pruner = TokenPruner(self.prune_ratio, prune_index)
230230

231231

0 commit comments

Comments
 (0)