Skip to content

Commit 99c25fa

Browse files
brianhou0208rwightman
authored andcommitted
fix nest type error
1 parent 72b2a09 commit 99c25fa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/nest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -449,7 +449,7 @@ def forward_intermediates(
449449

450450
# forward pass
451451
x = self.patch_embed(x)
452-
last_idx = self.num_blocks - 1
452+
last_idx = len(self.num_blocks) - 1
453453
if torch.jit.is_scripting() or not stop_early: # can't slice blocks in torchscript
454454
stages = self.levels
455455
else:

0 commit comments

Comments
 (0)