mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

fix nest type error

Browse Source
This commit is contained in:
Ryan 2025-05-06 03:43:49 +08:00 committed by Ross Wightman
parent 72b2a09106
commit 99c25fa5c0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/nest.py
View File

@ -449,7 +449,7 @@ class Nest(nn.Module):
# forward pass # forward pass
x = self.patch_embed(x) x = self.patch_embed(x)
last_idx = self.num_blocks - 1 last_idx = len(self.num_blocks) - 1
if torch.jit.is_scripting() or not stop_early: # can't slice blocks in torchscript if torch.jit.is_scripting() or not stop_early: # can't slice blocks in torchscript
stages = self.levels stages = self.levels
else: else: