Update davit.py

3 years ago · a813bbd1df
parent 62b9d696b0
commit a813bbd1df
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -514,7 +514,7 @@ class DaViT(nn.Module):
            features[-1], sizes[-1] = patch_layer(features[-1], sizes[-1])
            
            for block in enumerate(stage):
-                for layer in enumerate(block):
+                for _, layer in enumerate(block):
                    if self.grad_checkpointing and not torch.jit.is_scripting():
                        features[-1], sizes[-1] = checkpoint.checkpoint(layer, (features[-1], sizes[-1]))
                    else: